From 737084ec3d1c062ffa6de4fe27c1cfb6231fbf1b Mon Sep 17 00:00:00 2001
From: "Jovian Anthony Jaison (Meta Employee)"
 <38627145+jovianjaison@users.noreply.github.com>
Date: Wed, 2 Oct 2024 18:49:06 -0700
Subject: [PATCH] Add _dynamo.config inline_inbuilt_nn_modules and
 specialize_float logging (#137139)

Summary:
X-link: https://github.com/pytorch/pytorch/pull/137139
Approved by: https://github.com/ezyang

Reviewed By: PaliC

Differential Revision: D63783497

Pulled By: jovianjaison

fbshipit-source-id: 5abe70d558917a9807e33be8181d42ef240c5a95
---
 userbenchmark/dynamo/dynamobench/_dynamo/utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py
index 5257f0cce..7205fdaaf 100644
--- a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py
+++ b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py
@@ -805,6 +805,8 @@ class CompilationMetrics:
     remote_cache_time_saved_s: Optional[float]
     structured_logging_overhead_s: Optional[float]
     config_suppress_errors: Optional[bool]
+    config_inline_inbuilt_nn_modules: Optional[bool]
+    specialize_float: Optional[bool]
 
 
 @dataclasses.dataclass