Split tutorials to 3 groups

Signed-off-by: Pavel Chekin <[email protected]>

Author: pbchekin

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

@@ -299,10 +299,41 @@ jobs:
         run: |
           ${{ env.TRITON_TEST_CMD }} --interpreter
 
-      - name: Run Tutorials
+      # FIXME: make sure new tutorials are added to one of the groups (mxfp, scaled_dot, rest)
+
+      - name: Select tutorials to run (mxfp)
+        if: matrix.suite == 'mxfp'
+        run: |
+          cat <<EOF | tee tutorials.txt
+          06-fused-attention
+          EOF
+
+      - name: Select tutorials to run (scaled_dot)
+        if: matrix.suite == 'scaled_dot'
+        run: |
+          cat <<EOF | tee tutorials.txt
+          08-grouped-gemm
+          10-experimental-block-pointer
+          EOF
+
+      - name: Select tutorials to run (rest)
         if: matrix.suite == 'rest'
         run: |
-          ${{ env.TRITON_TEST_CMD }} --tutorial
+          cat <<EOF | tee tutorials.txt
+          10i-experimental-block-pointer
+          03-matrix-multiplication
+          03i-matrix-multiplication
+          05-layer-norm
+          02-fused-softmax
+          04-low-memory-dropout
+          01-vector-add
+          07-extern-functions
+          EOF
+
+      - name: Run Tutorials
+        if: matrix.suite == 'mxfp' || matrix.suite == 'scaled_dot' || matrix.suite == 'rest'
+        run: |
+          ${{ env.TRITON_TEST_CMD }} --select-from-file tutorials.txt --tutorial
 
       - name: Get transformers version
         if: matrix.suite == 'rest'

scripts/pytest-utils.sh

@@ -59,6 +59,10 @@ pytest() {
 }
 
 run_tutorial_test() {
+    if [[ -v TRITON_TEST_SELECTFILE ]] && ! grep -qF "$1" "$TRITON_TEST_SELECTFILE"; then
+        return
+    fi
+
     echo
     echo "****** Running $1 test ******"
     echo