Skip to content

Commit 34b4228

Browse files
committed
Extend list of small GPUs for run_triton_kernels_tests
1 parent 9e23713 commit 34b4228

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/test-triton.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -722,8 +722,8 @@ 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+
if [[ -f "$gpu_file" ]] && grep -Eq "(B580|64a0|A770)" "$gpu_file"; then
726+
# Using any other number of processes results in an error on small GPUs due to insufficient resources.
727727
# FIXME: reconsider in the future
728728
max_procs=1
729729
else

0 commit comments

Comments
 (0)