diff --git a/tritonbench/operators/fp8_gemm/fp8_gemm.py b/tritonbench/operators/fp8_gemm/fp8_gemm.py index e35e34bb8..7acf87c1d 100644 --- a/tritonbench/operators/fp8_gemm/fp8_gemm.py +++ b/tritonbench/operators/fp8_gemm/fp8_gemm.py @@ -17,6 +17,10 @@ from .tutorial import matmul as tutorial_matmul +torch._dynamo.config.recompile_limit = ( + 10000 # Set high recompile limit to allow for exhausting autotuning +) + logger = logging.getLogger(__name__) try: from .persistent import (