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 788795b commit 0581dcfCopy full SHA for 0581dcf
scripts/test-triton.sh
@@ -722,8 +722,9 @@ 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.
+ # BMG, LNL, ARLs, A770
+ 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.
728
# FIXME: reconsider in the future
729
max_procs=1
730
else
0 commit comments