Always recompile kernels in perf regression tests in case the user do…

…es no specify otherwise
ROCm · Oct 4, 2024 · ad1eb5a · ad1eb5a
1 parent f866bf1
commit ad1eb5a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/perf-kernels/tools/tune_gemm/test_regression.py b/python/perf-kernels/tools/tune_gemm/test_regression.py
@@ -1,5 +1,6 @@
 import tune_gemm
 
+import os
 import yaml
 import pytest
 import warnings
@@ -115,9 +116,10 @@ def test_matmul_performance_regression(self, config, record_property):
         bias_vector = config.get('bias_vector', False)
         bias_size = M if bias_vector else 0
 
-        tune_gemm.run_bash_command("rm -rf ~/.triton/cache")
-        tune_gemm.run_bash_command(f"rm -rf {tune_gemm.get_filename_myKernels()}")
+        # Always compile if the user did not specify
+        os.environ.setdefault('TRITON_ALWAYS_COMPILE', '1')
 
+        tune_gemm.run_bash_command(f"rm -rf {tune_gemm.get_filename_myKernels()}")
         tune_gemm.generate_matmul_kernels([runConfig])
 
         gpus = [0]