[fbsync] [ROCm] remove HCC references (#8070)

Reviewed By: vmoens Differential Revision: D50789090 fbshipit-source-id: 569878f95d9adc67f4d6f425cef0aa35bc4d4b4c
pytorch · Nov 13, 2023 · 235f632 · 235f632
1 parent d9a5ed1
commit 235f632
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;