We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 70146d5 commit 6b8f6cfCopy full SHA for 6b8f6cf
aten/src/ATen/native/cuda/MemoryAccess.cuh
@@ -546,7 +546,7 @@ inline int can_vectorize_up_to(array_t pointers) {
546
#else
547
c10::DeviceIndex curDevice = -1;
548
AT_CUDA_CHECK(c10::cuda::GetDevice(&curDevice));
549
- if (at::detail::getCUDAHooks().isGPUArch(curDevice, {"gfx942"})) {
+ if (at::detail::getCUDAHooks().isGPUArch({"gfx942"}, curDevice)) {
550
if (pointers[0] == pointers[1])
551
return 4;
552
}
0 commit comments