Skip to content

Commit d83e0d9

Browse files
committed
Revert "[ROCm] remove HCC references (pytorch#8070)"
This reverts commit a63f98a.
1 parent a8ebd0b commit d83e0d9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ namespace {
8282
const int kMaxParallelImgs = 32;
8383

8484
inline unsigned int GET_THREADS() {
85-
#ifdef WITH_HIP
85+
#ifdef __HIP_PLATFORM_HCC__
8686
return 256;
8787
#endif
8888
return 512;

0 commit comments

Comments
 (0)