Revert "[ROCm] remove HCC references (pytorch#8070)"

This reverts commit a63f98a.
NicolasHug · Oct 30, 2023 · d83e0d9 · d83e0d9
1 parent a8ebd0b
commit d83e0d9
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 WITH_HIP
+#ifdef __HIP_PLATFORM_HCC__
   return 256;
 #endif
   return 512;