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

Fix IValue from SymBool on big-endian system (#163647)

Browse Source 
Skip test_compiled_autograd_attribution on s390x

It fails both on s390x and x86_64 at least under some circumstances. Disable it for now until on s390x until it works reliably.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163647
Approved by: https://github.com/malfet
This commit is contained in:
Aleksei Nikiforov
2025-10-14 15:07:48 +00:00
committed by PyTorch MergeBot
parent fbe0d20a17
commit c733072874
3 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
torch/testing/_internal/common_device_type.py
View File

@ -34,6 +34,7 @@ from torch.testing._internal.common_utils import (
IS_MACOS,
is_privateuse1_backend_available,
IS_REMOTE_GPU,
IS_S390X,
IS_SANDCASTLE,
IS_WINDOWS,
NATIVE_DEVICES,
@ -1337,6 +1338,10 @@ def _has_sufficient_memory(device, size):
else:
effective_size = size
# don't try using all RAM on s390x, leave some for service processes
if IS_S390X:
effective_size = effective_size * 2
if psutil.virtual_memory().available < effective_size:
gc.collect()
return psutil.virtual_memory().available >= effective_size