pytorch
d49f6d55 - [DataLoader] Fix tempfile binding and removing for torch_shm_manager (#57566)

Commit
3 years ago
[DataLoader] Fix tempfile binding and removing for torch_shm_manager (#57566) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57566 Fix the problem that `tempfile` has never been deleted even after `torch_shm_manager` is destroyed. - The previous implementation has wrong path length for the Linux Socket. It leads to we lose the last character of the name of `tempfile` when bind the pathname to socket. At the end, we can not delete this file due to unexpected file name. - After we solve the racing problem by introducing a temporary directory, it becomes more dangerous since it prevents `torch_shm_manager` to delete directory as the tempfile persists in the temporary directory. Test Plan: Imported from OSS Reviewed By: VitalyFedyunin Differential Revision: D28202866 Pulled By: ejguan fbshipit-source-id: 912cfd8fec0cc309d47df223b2b0faa599c60799
Author
Parents
Loading