Skip to content

Commit 0581dcf

Browse files
authored
Extend list of small GPUs for run_triton_kernels_tests (#5426)
1 parent 788795b commit 0581dcf

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

scripts/test-triton.sh

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -722,8 +722,9 @@ run_triton_kernels_tests() {
722722

723723
# available after `capture_runtime_env` call
724724
gpu_file="$TRITON_TEST_REPORTS_DIR/gpu.txt"
725-
if [[ -f "$gpu_file" ]] && grep -q "B580" "$gpu_file"; then
726-
# Using any other number of processes results in an error on the BMG due to insufficient resources.
725+
# BMG, LNL, ARLs, A770
726+
if [[ -f "$gpu_file" ]] && grep -Eq "(B580|64a0|7d6|7d5|A770)" "$gpu_file"; then
727+
# Using any other number of processes results in an error on small GPUs due to insufficient resources.
727728
# FIXME: reconsider in the future
728729
max_procs=1
729730
else

0 commit comments

Comments
 (0)