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

Update flatbuffer test models to match pkl models (#93022)

Browse Source 
Also regenerate upgrader with

```
python torchgen/operator_versions/gen_mobile_upgraders.py
```

Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/93022
Approved by: https://github.com/tugsbayasgalan
This commit is contained in:
Han Qi
2023-01-26 21:17:53 +00:00
committed by PyTorch MergeBot
parent 68a49322e7
commit 1f352f7c1f
12 changed files with 151 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torchgen/operator_versions/gen_mobile_upgraders.py
View File

@ -384,7 +384,7 @@ def main() -> None:
for up in sorted_upgrader_list:
print("after sort upgrader : ", next(iter(up)))
pytorch_dir = Path(__file__).resolve().parents[3]
pytorch_dir = Path(__file__).resolve().parents[2]
upgrader_path = pytorch_dir / "torch" / "csrc" / "jit" / "mobile"
write_cpp(str(upgrader_path), sorted_upgrader_list)