diff --git a/fairscale/clib/fused_adam_cuda/fused_adam_cuda_kernel.cu b/fairscale/clib/fused_adam_cuda/fused_adam_cuda_kernel.cu index 6663ce651..5884033d9 100644 --- a/fairscale/clib/fused_adam_cuda/fused_adam_cuda_kernel.cu +++ b/fairscale/clib/fused_adam_cuda/fused_adam_cuda_kernel.cu @@ -8,7 +8,6 @@ #include #include "ATen/TensorUtils.h" #include "ATen/AccumulateType.h" -#include #include "multi_tensor_apply.cuh" #define BLOCK_SIZE 512 @@ -262,5 +261,5 @@ void fused_adam_cuda( throw "Parameters must be of type float or half"; } } - THCudaCheck(cudaGetLastError()); + C10_CUDA_CHECK(cudaGetLastError()); }