From 9db567f17d20f70307025de528baa2c6d118ca93 Mon Sep 17 00:00:00 2001
From: Justin Chu <justinchuby@users.noreply.github.com>
Date: Wed, 24 Jul 2024 20:02:03 +0000
Subject: [PATCH] [ONNX] Set dump_exported_program to True in bench (#131670)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131670
Approved by: https://github.com/titaiwangms
---
 benchmarks/dynamo/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/benchmarks/dynamo/common.py b/benchmarks/dynamo/common.py
index 8069ca11520e..24bc8045f1ca 100644
--- a/benchmarks/dynamo/common.py
+++ b/benchmarks/dynamo/common.py
@@ -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: