From 51e43b6143969ea6570d5873cdcf3e430ac9b73e Mon Sep 17 00:00:00 2001 From: anj-s <32556631+anj-s@users.noreply.github.com> Date: Tue, 19 Oct 2021 17:33:04 -0700 Subject: [PATCH] remove deprecated func (#818) --- fairscale/clib/fused_adam_cuda/fused_adam_cuda_kernel.cu | 3 +-- 1 file changed, 1 insertion(+), 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 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()); }