Deprecate THC/THCGeneral.h

marcmac · marcmac · commit a3b7a034024d · 2023-02-13T21:29:51.000-08:00
diff --git a/csrc/lamb/fused_lamb_cuda_kernel.cu b/csrc/lamb/fused_lamb_cuda_kernel.cu
@@ -8,7 +8,6 @@
 #include "ATen/cuda/CUDAContext.h"
 #include "ATen/cuda/detail/IndexUtils.cuh"
 //#include "ATen/Type.h"
-#include <THC/THCGeneral.h>
 #include "ATen/AccumulateType.h"
 
 #include <iostream>
@@ -462,7 +461,7 @@ void fused_lamb_cuda(at::Tensor& p,
                         lamb_coeff.data<scalar_t>());
             }));
     }
-    THCudaCheck(cudaGetLastError());
+    C10_CUDA_CHECK(cudaGetLastError());
 }
 
 // template __device__ void reduce_two_vectors_in_register<float,512>(float a, float b, float* g_a,

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,6 @@`
`8`	`8`	`#include "ATen/cuda/CUDAContext.h"`
`9`	`9`	`#include "ATen/cuda/detail/IndexUtils.cuh"`
`10`	`10`	`//#include "ATen/Type.h"`
`11`		`-#include <THC/THCGeneral.h>`
`12`	`11`	`#include "ATen/AccumulateType.h"`
`13`	`12`
`14`	`13`	`#include <iostream>`
`@@ -462,7 +461,7 @@ void fused_lamb_cuda(at::Tensor& p,`
`462`	`461`	`lamb_coeff.data<scalar_t>());`
`463`	`462`	`}));`
`464`	`463`	`}`
`465`		`- THCudaCheck(cudaGetLastError());`
	`464`	`+ C10_CUDA_CHECK(cudaGetLastError());`
`466`	`465`	`}`
`467`	`466`
`468`	`467`	`// template __device__ void reduce_two_vectors_in_register<float,512>(float a, float b, float* g_a,`