Use PYTEST_MAX_PROCESSES for parallel jobs number (#5438)

Fixes #5437

---------

Signed-off-by: Gregory Shimansky <[email protected]>

Author: gshimansky

.github/workflows/build-test-reusable.yml

@@ -338,6 +338,7 @@ jobs:
       - name: Run triton kernels tests
         if: matrix.suite == 'triton-kernels'
         run: |
+          export PYTEST_MAX_PROCESSES=4
           ${{ env.TRITON_TEST_CMD }} --triton-kernels
 
       # FIXME: make sure new tutorials are added to one of the groups (scaled_dot, rest, tutorial-faX)

scripts/test-triton.sh

@@ -730,7 +730,7 @@ run_triton_kernels_tests() {
   else
     # Using any other number of processes results in an error on the PVC due to insufficient resources.
     # FIXME: reconsider in the future
-    max_procs=4
+    max_procs=${PYTEST_MAX_PROCESSES:-4}
   fi
   # skipping mxfp, they are part of mxfp_tests suite
   TRITON_TEST_SUITE=triton_kernels \