Skip to content

Commit bf18cc9

Browse files
oulgenfacebook-github-bot
authored andcommitted
Rename cache limit to recompile limit in configs (#143709)
Summary: This PR renames every cache_limit to recompile_limit via sed. Old config options are maintained via Config(alias='xyz') X-link: pytorch/pytorch#143709 Approved by: https://github.com/jansel Reviewed By: aorenste Differential Revision: D67580275 Pulled By: oulgen fbshipit-source-id: 5f44f01a2d43ee280a4872b112aff7eaf7b71700
1 parent 07e6ef4 commit bf18cc9

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2401,16 +2401,16 @@ def format_func_info(code):
24012401

24022402
@contextlib.contextmanager
24032403
def disable_cache_limit():
2404-
prior = config.cache_size_limit
2405-
config.cache_size_limit = sys.maxsize
2406-
prior_acc_limit = config.accumulated_cache_size_limit
2407-
config.accumulated_cache_size_limit = sys.maxsize
2404+
prior = config.recompile_limit
2405+
config.recompile_limit = sys.maxsize
2406+
prior_acc_limit = config.accumulated_recompile_limit
2407+
config.accumulated_recompile_limit = sys.maxsize
24082408

24092409
try:
24102410
yield
24112411
finally:
2412-
config.cache_size_limit = prior
2413-
config.accumulated_cache_size_limit = prior_acc_limit
2412+
config.recompile_limit = prior
2413+
config.accumulated_recompile_limit = prior_acc_limit
24142414

24152415

24162416
# map from transformed code back to original user code

userbenchmark/dynamo/dynamobench/torchao_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ def setup_baseline():
88

99
recommended_inductor_config_setter()
1010
torch._dynamo.config.automatic_dynamic_shapes = False
11-
torch._dynamo.config.cache_size_limit = 10000
11+
torch._dynamo.config.recompile_limit = 10000
1212

1313

1414
def torchao_optimize_ctx(quantization: str):

0 commit comments

Comments
 (0)