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

Revert "Support IPC for Expandable Segments (#130890)"

Browse Source 
This reverts commit 32c2f84e349ad6e34b8559d3f1f9c27020ae702f.

Reverted https://github.com/pytorch/pytorch/pull/130890 on behalf of https://github.com/zdevito due to variable shadowing broke internal tests ([comment](https://github.com/pytorch/pytorch/pull/130890#issuecomment-2245456085))
This commit is contained in:
PyTorch MergeBot
2024-07-23 14:46:27 +00:00
parent f064dac588
commit 1e86387871
4 changed files with 62 additions and 239 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_multiprocessing.py
View File

@ -48,6 +48,9 @@ TEST_CUDA_IPC = (
TEST_MULTIGPU = TEST_CUDA_IPC and torch.cuda.device_count() > 1
if TEST_CUDA_IPC:
torch.cuda.memory._set_allocator_settings("expandable_segments:False")
class SubProcess(mp.Process):
def __init__(self, tensor):