[ROCm] remove HCC references (#8070)

a63f98ad · Jeff Daily · GitHub · 3fb88b3e · a63f98ad
Unverified Commit a63f98ad authored Oct 27, 2023 by Jeff Daily Committed by GitHub Oct 27, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu +1 -1

No files found.
--- 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;