Enable FX graph caching for all benchmarks

Summary: Test Plan: ghstack-source-id: 1ff0b8540be01816b0e3cfb03b059a0412f8bb55 Pull Request resolved: #125980
pytorch · May 14, 2024 · 2bde357 · 2bde357
1 parent aac215a
commit 2bde357
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/benchmarks/dynamo/common.py b/benchmarks/dynamo/common.py
@@ -103,6 +103,9 @@
 # Suppress torch.profiler spam
 os.environ["KINETO_LOG_LEVEL"] = "5"
 
+# Enable FX graph caching
+os.environ["TORCHINDUCTOR_FX_GRAPH_CACHE"] = "1"
+
 current_name = ""
 current_device = ""
 current_onnx_compiler = ""