Skip to content

Commit 6b8f6cf

Browse files
committed
pulled latest changes, one PR made in between was using the old signature.
1 parent 70146d5 commit 6b8f6cf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

aten/src/ATen/native/cuda/MemoryAccess.cuh

+1-1
Original file line numberDiff line numberDiff line change
@@ -546,7 +546,7 @@ inline int can_vectorize_up_to(array_t pointers) {
546546
#else
547547
c10::DeviceIndex curDevice = -1;
548548
AT_CUDA_CHECK(c10::cuda::GetDevice(&curDevice));
549-
if (at::detail::getCUDAHooks().isGPUArch(curDevice, {"gfx942"})) {
549+
if (at::detail::getCUDAHooks().isGPUArch({"gfx942"}, curDevice)) {
550550
if (pointers[0] == pointers[1])
551551
return 4;
552552
}

0 commit comments

Comments
 (0)