frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use linux.g4dn.4xlarge.nvidia.gpu for cuda 12.4 legacy driver tests (#163956)

Browse Source 
Workaround for https://github.com/pytorch/pytorch/issues/163658

Looks like the workflow passes on 12.8 build that use inux.g4dn.4xlarge.nvidia.gpu but its failing on 12.6 builds that use linux.4xlarge.nvidia.gpu: https://github.com/pytorch/pytorch/actions/runs/17953843505/job/51080623612#step:13:470

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163956
Approved by: https://github.com/malfet

Co-authored-by: Mark Saroufim <marksaroufim@meta.com>
This commit is contained in:
atalman
2025-09-29 19:38:17 +00:00
committed by PyTorch MergeBot
parent f090818a40
commit 349c960970
2 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_cuda.py
View File

@ -7296,9 +7296,7 @@ class TestCudaDeviceParametrized(TestCase):
"""
from torch.cuda import _compile_kernel
spin_wait_kernel = _compile_kernel(
kernel_source, "wait_for_cpu", compute_capability="70"
)
spin_wait_kernel = _compile_kernel(kernel_source, "wait_for_cpu")
x = torch.ones(4, device="cuda")
x_cpu = torch.zeros(x.shape, device="cpu").pin_memory()