Skip to content

GitLab

Unverified Commit 51e43b61 authored by anj-s's avatar anj-s Committed by GitHub
Browse files

remove deprecated func (#818)

parent fb7b6a93
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
fairscale/clib/fused_adam_cuda/fused_adam_cuda_kernel.cu
View file @ 51e43b61
...@@ -8,7 +8,6 @@ ...@@ -8,7 +8,6 @@
#include <cmath> #include <cmath>
#include "ATen/TensorUtils.h" #include "ATen/TensorUtils.h"
#include "ATen/AccumulateType.h" #include "ATen/AccumulateType.h"
#include <THC/THCGeneral.h>
#include "multi_tensor_apply.cuh" #include "multi_tensor_apply.cuh"
#define BLOCK_SIZE 512 #define BLOCK_SIZE 512
...@@ -262,5 +261,5 @@ void fused_adam_cuda( ...@@ -262,5 +261,5 @@ void fused_adam_cuda(
throw "Parameters must be of type float or half"; throw "Parameters must be of type float or half";
} }
} }
THCudaCheck(cudaGetLastError()); C10_CUDA_CHECK(cudaGetLastError());
} }
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment