[CI] Fix triton_kernels hang

Signed-off-by: Whitney Tsang <[email protected]>

Author: whitneywhtsang

scripts/test-triton.sh

@@ -721,7 +721,7 @@ run_triton_kernels_tests() {
 
   # available after `capture_runtime_env` call
   gpu_file="$TRITON_TEST_REPORTS_DIR/gpu.txt"
-  if [[ -f "$gpu_file" ]] && grep -q "B580" "$gpu_file"; then
+  if [[ -f "$gpu_file" ]] && grep -qE "B580|A770" "$gpu_file"; then
     # Using any other number of processes results in an error on the BMG due to insufficient resources.
     # FIXME: reconsider in the future
     max_procs=1