frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-12 06:44:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

[cutlass backend][BE] remove force disable cache in tests (#158053)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/158053
Approved by: https://github.com/coconutruben
This commit is contained in:
henrylhtsang
2025-07-15 01:55:15 +00:00
committed by PyTorch MergeBot
parent e40ade5182
commit 6200584193
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/inductor/test_cutlass_backend.py
View File

@ -421,7 +421,9 @@ class TestCutlassBackend(TestCase):
2, 2,
4, 4,
], # guarantees > 1 choices ], # guarantees > 1 choices
"force_disable_caches": True, "fx_graph_cache": False,
"fx_graph_remote_cache": False,
"autotune_local_cache": False,
} }
): ):
from torch._inductor.utils import run_and_get_code from torch._inductor.utils import run_and_get_code
@ -1530,7 +1532,8 @@ class TestCutlassBackend(TestCase):
"max_autotune_gemm_backends": "ATEN,TRITON,CUTLASS", "max_autotune_gemm_backends": "ATEN,TRITON,CUTLASS",
"cuda.cutlass_max_profiling_configs": 2, "cuda.cutlass_max_profiling_configs": 2,
# needed for log searching # needed for log searching
"force_disable_caches": True, "fx_graph_cache": False,
"fx_graph_remote_cache": False,
} }
): ):
with ( with (