Skip to content

Commit 83f3253

Browse files
authored
Do not crash autotuner on more triton/llvm/cuda errors seen on B200 (#374)
1 parent 27b2774 commit 83f3253

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

helion/autotuner/base_search.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,16 @@
4343
from . import ConfigSpec
4444

4545
_expected_errors_regexp: re.Pattern[str] = re.compile(
46-
r"|".join(map(re.escape, ["[CUDA]: invalid argument"]))
46+
r"|".join(
47+
map(
48+
re.escape,
49+
[
50+
"[CUDA]: invalid argument", # CUDA Error
51+
"misaligned address", # CUDA Error
52+
"PassManager::run failed", # Triton Error
53+
],
54+
)
55+
)
4756
)
4857

4958

0 commit comments

Comments
 (0)