diff --git a/helion/autotuner/base_search.py b/helion/autotuner/base_search.py index 365ac2848..991592827 100644 --- a/helion/autotuner/base_search.py +++ b/helion/autotuner/base_search.py @@ -43,7 +43,16 @@ from . import ConfigSpec _expected_errors_regexp: re.Pattern[str] = re.compile( - r"|".join(map(re.escape, ["[CUDA]: invalid argument"])) + r"|".join( + map( + re.escape, + [ + "[CUDA]: invalid argument", # CUDA Error + "misaligned address", # CUDA Error + "PassManager::run failed", # Triton Error + ], + ) + ) )