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

Expose more headers for extensions. (#95447)

Browse Source 
Fixes #ISSUE_NUMBER

Expose more headers for extensions of distributed methods.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/95447
Approved by: https://github.com/ezyang
This commit is contained in:
donnyyou
2023-02-27 18:59:36 +00:00
committed by PyTorch MergeBot
parent c1fa403e57
commit 5d70ee93fa
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@ -1158,6 +1158,9 @@ def main():
'include/torch/csrc/distributed/c10d/*.h',
'include/torch/csrc/distributed/c10d/*.hpp',
'include/torch/csrc/distributed/rpc/*.h',
'include/torch/csrc/distributed/autograd/context/*.h',
'include/torch/csrc/distributed/autograd/functions/*.h',
'include/torch/csrc/distributed/autograd/rpc_messages/*.h',
'include/torch/csrc/jit/*.h',
'include/torch/csrc/jit/backends/*.h',
'include/torch/csrc/jit/generated/*.h',