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

[ONNX] Set dump_exported_program to True in bench (#131670)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131670
Approved by: https://github.com/titaiwangms
This commit is contained in:
Justin Chu
2024-07-24 20:02:03 +00:00
committed by PyTorch MergeBot
parent 85fa66be04
commit 9db567f17d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/dynamo/common.py
View File

@ -1690,7 +1690,7 @@ class OnnxModelFromTorchScript(OnnxModel):
torch_onnx.patch_torch(
error_report=True,
profile=True,
dump_exported_program=False,
dump_exported_program=True,
artifacts_dir=os.path.dirname(output_path),
)
else: