OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 52 Packages Projects Releases Wiki Activity

Add basic OpenReg module scaffolding with autograd (#131708)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131708
Approved by: https://github.com/ezyang
This commit is contained in:
albanD
2024-08-02 17:17:55 -04:00
committed by PyTorch MergeBot
parent df59084012
commit 3d87dfc088
8 changed files with 435 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.flake8
View File

@@ -55,6 +55,9 @@ per-file-ignores =
torch/distributed/_functional_collectives.py: TOR901
torch/distributed/_spmd/data_parallel.py: TOR901
torch/distributed/_tensor/_collective_utils.py: TOR901
# This is a full package that happen to live within the test
# folder, so ok to skip
test/cpp_extensions/open_registration_extension/pytorch_openreg/__init__.py: TOR901
optional-ascii-coding = True
exclude =
./.git,