saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[ci, 3.13] disable failing cpp_extension test due to weights_only error in numpy 2.1 (#141623)

Browse source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/141623
Approved by: https://github.com/mikaylagawarecki, https://github.com/atalman
ghstack dependencies: #141409, #142003, #141572, #141577, #141605, #141621
This commit is contained in:
William Wen 2024-12-04 00:02:47 -08:00 committed by PyTorch MergeBot
parent 2be8d16247
commit cd56cd30f2
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
test/test_cpp_extensions_open_device_registration.py
View file

@ -534,6 +534,10 @@ class TestCppExtensionOpenRgistration(common.TestCase):
np.__version__ < "1.25",
"versions < 1.25 serialize dtypes differently from how it's serialized in data_legacy_numpy",
)
@unittest.skipIf(
np.__version__ >= "2.1",
"weights_only failure on numpy >= 2.1",
)
def test_open_device_numpy_serialization(self):
"""
This tests the legacy _rebuild_device_tensor_from_numpy serialization path