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

Smoke tests don't run nvshmem on Windows (#162646)

Browse Source 
Only available for linux x86 and aarch64 :
https://pypi.org/project/nvidia-nvshmem-cu13/#files

nvshmem is available only on linux:
``
"nvidia-nvshmem-cu12==3.3.24; platform_system == 'Linux' and platform_machine == 'x86_64' | "
``
https://github.com/pytorch/pytorch/blob/main/.github/scripts/generate_binary_build_matrix.py#L57
Pull Request resolved: https://github.com/pytorch/pytorch/pull/162646
Approved by: https://github.com/kwen2501
This commit is contained in:
atalman
2025-09-11 16:09:20 +00:00
committed by PyTorch MergeBot
parent 24492cbab2
commit 2f5a24c2a2
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
.ci/pytorch/smoke_test/smoke_test.py
View File

@ -386,8 +386,8 @@ def smoke_test_compile(device: str = "cpu") -> None:
def smoke_test_nvshmem() -> None: def smoke_test_nvshmem() -> None:
if not torch.cuda.is_available(): if not torch.cuda.is_available() or target_os == "windows":
print("CUDA is not available, skipping NVSHMEM test") print("Windows platform or CUDA is not available, skipping NVSHMEM test")
return return
# Check if NVSHMEM is compiled in current build # Check if NVSHMEM is compiled in current build
@ -396,7 +396,9 @@ def smoke_test_nvshmem() -> None:
except ImportError: except ImportError:
# Not built with NVSHMEM support. # Not built with NVSHMEM support.
# torch is not compiled with NVSHMEM prior to 2.9 # torch is not compiled with NVSHMEM prior to 2.9
if torch.__version__ < "2.9": from torch.torch_version import TorchVersion
if TorchVersion(torch.__version__) < (2, 9):
return return
else: else:
# After 2.9: NVSHMEM is expected to be compiled in current build # After 2.9: NVSHMEM is expected to be compiled in current build