[CI] Enable matrix test for UT (#1677)

Enable matrix test method to accelerate the UT test.

Author: RUIJIEZHONG66166

.github/workflows/_linux_build.yml

@@ -28,6 +28,11 @@ on:
         type: string
         default: 'linux.idc.xpu'
         description: Runner label
+      triton:
+        required: false
+        type: string
+        default: ''
+        description: Triton commit. Use pytorch pined commit by default
     outputs:
       torch_commit_id:
         description: The commit id of the torch build
@@ -70,6 +75,30 @@ jobs:
         uses: actions/checkout@v4
         with:
           path: torch-xpu-ops
+      - name: Build Triton XPU
+        run: |
+          # gcc 13
+          dnf install -y gcc-toolset-13-gcc-c++
+          source /opt/rh/gcc-toolset-13/enable
+          dnf install -y zlib-devel
+          cd ../ && rm -rf pytorch
+          git clone https://github.com/pytorch/pytorch pytorch
+          cd pytorch
+          if [ -z ${{ inputs.triton }} ]; then
+            TRITON_COMMIT_ID="$(<.ci/docker/ci_commit_pins/triton-xpu.txt)"
+          else
+            TRITON_COMMIT_ID="${{ inputs.triton }}"
+          fi
+          if [ "${{ inputs.pytorch }}" != "nightly_wheel" ]; then
+            pip install cmake ninja pybind11
+            rm -rf pytorch_triton_xpu-*.whl
+            TRITON_VERSION_NAME="$(
+              curl -sSL https://raw.githubusercontent.com/intel/intel-xpu-backend-for-triton/${TRITON_COMMIT_ID}/python/triton/__init__.py 2>&1 |\
+                      grep '__version__' |head -n 1 |awk -F "'" '{print $2}'
+            )"
+            python .github/scripts/build_triton_wheel.py --device xpu --commit-hash ${TRITON_COMMIT_ID} --triton-version ${TRITON_VERSION_NAME}
+            cp pytorch_triton_xpu-*.whl ${{ github.workspace }}
+          fi
       - name: Build Pytorch XPU
         run: |
           set -xe -o pipefail
@@ -118,6 +147,12 @@ jobs:
         with:
           name: Torch-XPU-Wheel-${{ github.event.pull_request.number || github.sha }}
           path: ${{ github.workspace }}/torch*.whl
+      - name: Upload Triton Wheel
+        if: ${{ ! cancelled() }}
+        uses: actions/upload-artifact@v4
+        with:
+          name: Triton-Wheel-${{ github.event.pull_request.number || github.sha }}
+          path: ${{ github.workspace }}/pytorch_triton_xpu-*.whl
       - name: Upload Build Log
         if: ${{ ! cancelled() }}
         uses: actions/upload-artifact@v4
@@ -128,4 +163,4 @@ jobs:
         if: always()
         run: |
           chmod 777 . -R
-          rm -rf pytorch torch-xpu-ops pytorch_*.log torch*.whl
+          rm -rf pytorch torch-xpu-ops pytorch_*.log torch*.whl pytorch_triton_xpu-*.whl