frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Rename cache limit to recompile limit in configs (#143709)

Browse Source 
This PR renames every cache_limit to recompile_limit via sed.

Old config options are maintained via Config(alias='xyz')

Pull Request resolved: https://github.com/pytorch/pytorch/pull/143709
Approved by: https://github.com/jansel
This commit is contained in:
Oguz Ulgen
2024-12-21 21:45:17 -08:00
committed by PyTorch MergeBot
parent 9bf4b1c2e9
commit dc55704b48
23 changed files with 95 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/transformer/score_mod.py
View File

@ -26,7 +26,7 @@ from torch.nn.attention.flex_attention import (
torch._dynamo.config.automatic_dynamic_shapes = False
# Needed since changing args to function causes recompiles
torch._dynamo.config.cache_size_limit = 1000
torch._dynamo.config.recompile_limit = 1000
from torch._inductor.runtime.benchmarking import benchmarker