Skip to content

Commit 52df740

Browse files
authored
Use broader A770 name when selecting max_procs (#5449)
1 parent 88a07f2 commit 52df740

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
@@ -723,7 +723,7 @@ run_triton_kernels_tests() {
723723
# available after `capture_runtime_env` call
724724
gpu_file="$TRITON_TEST_REPORTS_DIR/gpu.txt"
725725
# BMG, LNL, ARLs, A770
726-
if [[ -f "$gpu_file" ]] && grep -Eq "(B580|64a0|7d6|7d5|A770)" "$gpu_file"; then
726+
if [[ -f "$gpu_file" ]] && grep -Eq "(B580|64a0|7d6|7d5|770)" "$gpu_file"; then
727727
# Using any other number of processes results in an error on small GPUs due to insufficient resources.
728728
# FIXME: reconsider in the future
729729
max_procs=1

0 commit comments

Comments
 (0)