Skip to content

Commit e395a79

Browse files
authored
Fix DeformConvTester::test_backward_cuda by setting threads per block to 512 (#3942)
1 parent 2cc8359 commit e395a79

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,10 +85,7 @@ inline unsigned int GET_THREADS() {
8585
#ifdef __HIP_PLATFORM_HCC__
8686
return 256;
8787
#endif
88-
if (at::cuda::getCurrentDeviceProperties()->major >= 6) {
89-
return 1024;
90-
}
91-
return 512;
88+
return 512;
9289
}
9390

9491
inline unsigned int GET_BLOCKS(

0 commit comments

Comments
 (0)