frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[EASY] Quick Fix for broken shape function autogen.

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76703

Approved by: https://github.com/eellison
This commit is contained in:
John Clow
2022-05-02 16:42:51 -07:00
committed by PyTorch MergeBot
parent ccd5fa506f
commit db21e22b4b
2 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torchgen/shape_functions/gen_jit_shape_functions.py
View File

@ -114,7 +114,7 @@ def write_decomposition_util_file(path: str) -> None:
def main() -> None:
pytorch_dir = Path(__file__).resolve().parents[3]
pytorch_dir = Path(__file__).resolve().parents[2]
upgrader_path = pytorch_dir / "torch" / "csrc" / "jit" / "runtime"
write_decomposition_util_file(str(upgrader_path))