Skip to content

Commit 84abd93

Browse files
committed
Extend list of small GPUs for run_triton_kernels_tests
1 parent 34b4228 commit 84abd93

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/test-triton.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -722,7 +722,7 @@ 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 -Eq "(B580|64a0|A770)" "$gpu_file"; then
725+
if [[ -f "$gpu_file" ]] && grep -Eq "(B580|64a0|7d6|7d5|A770)" "$gpu_file"; then
726726
# 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

0 commit comments

Comments
 (0)