Limit test

Author: whitneywhtsang

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

@@ -278,90 +278,11 @@ jobs:
         run: |
           echo "TRITON_TEST_CMD=${{ needs.build.outputs.test-triton-command }}" | tee -a $GITHUB_ENV
 
-      - name: Run Proton tests
-        if: matrix.suite == 'rest' && inputs.driver_version == 'rolling' && inputs.device == 'max1100'
-        run: |
-          cd third_party/proton/test
-          pytest test_api.py test_lib.py test_profile.py test_viewer.py test_record.py -s -v
-          cd ..
-
-      - name: Run minicore tests
-        if: matrix.suite == 'minicore'
-        run: |
-          ${{ env.TRITON_TEST_CMD }} --minicore
-
-      - name: Run mxfp tests
-        if: matrix.suite == 'mxfp'
-        run: |
-          ${{ env.TRITON_TEST_CMD }} --mxfp
-
       - name: Run scaled_dot tests
         if: matrix.suite == 'scaled_dot'
         run: |
           ${{ env.TRITON_TEST_CMD }} --scaled-dot
 
-      - name: Run interpreter tests
-        if: matrix.suite == 'rest'
-        run: |
-          ${{ env.TRITON_TEST_CMD }} --interpreter
-
-      # 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
-          09-persistent-matmul
-          EOF
-
-      - name: Select tutorials to run (scaled_dot)
-        if: matrix.suite == 'scaled_dot'
-        run: |
-          cat <<EOF | tee tutorials.txt
-          06-fused-attention
-          EOF
-
-      - name: Select tutorials to run (rest)
-        if: matrix.suite == 'rest'
-        run: |
-          cat <<EOF | tee tutorials.txt
-          01-vector-add
-          02-fused-softmax
-          03-matrix-multiplication
-          04-low-memory-dropout
-          05-layer-norm
-          07-extern-functions
-          08-grouped-gemm
-          10-experimental-block-pointer
-          10i-experimental-block-pointer
-          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'
-        run: |
-          cd pytorch
-          TRANSFORMERS_VERSION="$(<.ci/docker/ci_commit_pins/huggingface.txt)"
-          echo "TRANSFORMERS_VERSION=$TRANSFORMERS_VERSION" | tee -a $GITHUB_ENV
-
-      - name: Install transformers
-        if: matrix.suite == 'rest'
-        uses: ./.github/actions/install-dependency
-        with:
-          package: transformers
-          repository: huggingface/transformers
-          ref: ${{ env.TRANSFORMERS_VERSION }}
-          try-tag-prefix: v
-
-      - name: Run E2E test
-        if: matrix.suite == 'rest'
-        run: |
-          timeout -s KILL 4900 ${{ env.TRITON_TEST_CMD }} --inductor || ${{ inputs.ignore_errors }}
-
       - name: Save pip cache
         if: ${{ steps.pip-cache.outputs.status == 'miss' }}
         uses: ./.github/actions/save

scripts/test-triton.sh

@@ -326,8 +326,7 @@ run_scaled_dot_tests() {
   cd $TRITON_PROJ/python/test/unit
 
   TRITON_DISABLE_LINE_INFO=1 TRITON_TEST_SUITE=scaled_dot \
-    run_pytest_command -vvv -n ${PYTEST_MAX_PROCESSES:-8} --device xpu language/ --ignore=language/test_line_info.py --ignore=language/test_subprocess.py --ignore=language/test_warp_specialization.py --ignore=language/test_frontend.py\
-    -k "test_scaled_dot"
+    run_pytest_command -vvv -n ${PYTEST_MAX_PROCESSES:-8} --device xpu language/test_core.py -k "test_scaled_dot"
 }
 
 run_core_tests() {