2023-10-27 nightly release (a63f98a)

pytorch · Oct 27, 2023 · 36aab13 · 36aab13
1 parent bb034e0
commit 36aab13
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;