fix GET_THREADS() for ROCm (#2997)

ashishfarmer · web-flow · commit dbdc72671166 · 2020-11-16T16:06:28.000+01:00
diff --git a/torchvision/csrc/cuda/DeformConv_cuda.cu b/torchvision/csrc/cuda/DeformConv_cuda.cu
@@ -81,6 +81,9 @@
 const int kMaxParallelImgs = 32;
 
 inline unsigned int GET_THREADS() {
+#ifdef __HIP_PLATFORM_HCC__
+  return 256;
+#endif
   if (at::cuda::getCurrentDeviceProperties()->major >= 6) {
     return 1024;
   }

Original file line number	Diff line number	Diff line change
`@@ -81,6 +81,9 @@`
`81`	`81`	`const int kMaxParallelImgs = 32;`
`82`	`82`
`83`	`83`	`inline unsigned int GET_THREADS() {`
	`84`	`+#ifdef __HIP_PLATFORM_HCC__`
	`85`	`+ return 256;`
	`86`	`+#endif`
`84`	`87`	`if (at::cuda::getCurrentDeviceProperties()->major >= 6) {`
`85`	`88`	`return 1024;`
`86`	`89`	`}`