We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9e23713 commit 34b4228Copy full SHA for 34b4228
scripts/test-triton.sh
@@ -722,8 +722,8 @@ run_triton_kernels_tests() {
722
723
# available after `capture_runtime_env` call
724
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.
+ if [[ -f "$gpu_file" ]] && grep -Eq "(B580|64a0|A770)" "$gpu_file"; then
+ # Using any other number of processes results in an error on small GPUs due to insufficient resources.
727
# FIXME: reconsider in the future
728
max_procs=1
729
else
0 commit comments