frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-06 09:17:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Multiprocessing support for NT (#110292)

Browse Source 
Fixes #110161

Allows NTs to be used in DataLoaders with `num_workers > 1`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110292
Approved by: https://github.com/cpuhrsch, https://github.com/albanD
This commit is contained in:
Joel Schlosser
2023-10-10 14:22:52 -04:00
committed by PyTorch MergeBot
parent 7f2d25c547
commit 43ea782af3
4 changed files with 100 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/autograd/gen_python_functions.py
View File

@ -159,8 +159,6 @@ _SKIP_PYTHON_BINDINGS = [
"fill.Scalar", # only used by the functionalization pass
"lift.*",
"normal_functional", # only used by the functionalization pas
"_nested_view_from_buffer_copy",
"_nested_view_from_buffer_copy_out",
"nbytes",
"itemsize",
]