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 Storage.filename to not track the filename when storage was mmap-ed with MAP_PRIVATE (#128725)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/128725
Approved by: https://github.com/albanD
This commit is contained in:
Mikayla Gawarecki
2024-06-17 08:55:40 -07:00
committed by PyTorch MergeBot
parent 213eba7d2e
commit b181b58857
4 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/test_tensor_creation_ops.py
View File

@ -3177,18 +3177,19 @@ class TestTensorCreation(TestCase):
dtype = torch.float64
t = torch.randn(2, 5, dtype=dtype, device=device)
with tempfile.NamedTemporaryFile() as f:
expected_filename = f.name if shared else None
t.numpy().tofile(f)
t_mapped = torch.from_file(f.name, shared=shared, size=t.numel(), dtype=dtype)
self.assertTrue(t_mapped.storage().filename == f.name)
self.assertTrue(t_mapped.untyped_storage().filename == expected_filename)
self.assertEqual(torch.flatten(t), t_mapped)
s = torch.UntypedStorage.from_file(f.name, shared, t.numel() * dtype.itemsize)
self.assertTrue(s.filename == f.name)
self.assertTrue(s.filename == expected_filename)
@onlyCPU
def test_storage_filename(self, device):
t = torch.randn(2, 5, device=device)
self.assertIsNone(t.storage().filename)
self.assertIsNone(t.untyped_storage().filename)
# Class for testing random tensor creation ops, like torch.randint