remove deprecated func (#818)

facebookresearch · Oct 20, 2021 · 51e43b6 · 51e43b6
1 parent fb7b6a9
commit 51e43b6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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 <cmath>
 #include "ATen/TensorUtils.h"
 #include "ATen/AccumulateType.h"
-#include <THC/THCGeneral.h>
 #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());
 }