Skip to content

Commit

Permalink
Set dump_exported_program to True in bench (#131670)
Browse files Browse the repository at this point in the history
Summary:
X-link: pytorch/pytorch#131670
Approved by: https://github.com/titaiwangms

Reviewed By: clee2000

Differential Revision: D60211332

fbshipit-source-id: 73645815dd998da88a0a3731dd0dcad392bf3b96
  • Loading branch information
justinchuby authored and facebook-github-bot committed Jul 25, 2024
1 parent fe36229 commit 06cf234
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion userbenchmark/dynamo/dynamobench/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -1690,7 +1690,7 @@ def forward(self, *args):
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:
Expand Down

0 comments on commit 06cf234

Please sign in to comment.