[ROCm] remove HCC references (#8070)

pytorch · Oct 27, 2023 · a63f98a · a63f98a
1 parent 3fb88b3
commit a63f98a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu b/torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu
@@ -82,7 +82,7 @@ namespace {
 const int kMaxParallelImgs = 32;
 
 inline unsigned int GET_THREADS() {
-#ifdef __HIP_PLATFORM_HCC__
+#ifdef WITH_HIP
   return 256;
 #endif
   return 512;