remove deprecated func (#818)

51e43b61 · anj-s · GitHub · fb7b6a93 · 51e43b61
Unverified Commit 51e43b61 authored Oct 19, 2021 by anj-s Committed by GitHub Oct 19, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

fairscale/clib/fused_adam_cuda/fused_adam_cuda_kernel.cu fairscale/clib/fused_adam_cuda/fused_adam_cuda_kernel.cu +1 -2

No files found.
--- 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());
 }