diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py index 742e1893d..1f8bca919 100644 --- a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py +++ b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py @@ -2401,16 +2401,16 @@ def format_func_info(code): @contextlib.contextmanager def disable_cache_limit(): - prior = config.cache_size_limit - config.cache_size_limit = sys.maxsize - prior_acc_limit = config.accumulated_cache_size_limit - config.accumulated_cache_size_limit = sys.maxsize + prior = config.recompile_limit + config.recompile_limit = sys.maxsize + prior_acc_limit = config.accumulated_recompile_limit + config.accumulated_recompile_limit = sys.maxsize try: yield finally: - config.cache_size_limit = prior - config.accumulated_cache_size_limit = prior_acc_limit + config.recompile_limit = prior + config.accumulated_recompile_limit = prior_acc_limit # map from transformed code back to original user code diff --git a/userbenchmark/dynamo/dynamobench/torchao_backend.py b/userbenchmark/dynamo/dynamobench/torchao_backend.py index 385485378..17876005a 100644 --- a/userbenchmark/dynamo/dynamobench/torchao_backend.py +++ b/userbenchmark/dynamo/dynamobench/torchao_backend.py @@ -8,7 +8,7 @@ def setup_baseline(): recommended_inductor_config_setter() torch._dynamo.config.automatic_dynamic_shapes = False - torch._dynamo.config.cache_size_limit = 10000 + torch._dynamo.config.recompile_limit = 10000 def torchao_optimize_ctx(quantization: str):