[CI/CD] Add test with FlagGems

.github/workflows/aipu-build-and-test.yml

@@ -89,3 +89,13 @@ jobs:
           python3.10 third_party/aipu/python/test/var_mean.py
           python3.10 third_party/aipu/python/test/var_mean_shared_memory.py
           #python3.10 third_party/aipu/python/test/var_mean_dma.py
+
+      - name: FlagTree Test with FlagGems
+        shell: bash
+        run: |
+          cd ../..
+          rm -rf FlagGems
+          git clone https://github.com/flagos-ai/FlagGems.git
+          cd FlagGems
+          git checkout addflagtreetest
+          python3.10 -m pytest -s --mode=quick --limit-cases=1 --skipped-spectest

.github/workflows/ascend-build-and-test.yml

@@ -76,3 +76,13 @@ jobs:
           python3 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized-flagtree.py
           python3 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized.py
           python3 third_party/ascend/examples/tutorials/14-accuracy-comparison.py
+
+      - name: FlagTree Test with FlagGems
+        shell: bash
+        run: |
+          cd ../..
+          rm -rf FlagGems
+          git clone https://github.com/flagos-ai/FlagGems.git
+          cd FlagGems
+          git checkout addflagtreetest
+          python3 -m pytest -s --mode=quick --limit-cases=1 --skipped-spectest

.github/workflows/cambricon-build-and-test.yml

@@ -60,3 +60,13 @@ jobs:
           export FLAGTREE_BACKEND=cambricon
           cd python
           MAX_JOBS=8 python3 -m pip install . --no-build-isolation
+
+      - name: FlagTree Test with FlagGems
+        shell: bash
+        run: |
+          cd ../..
+          rm -rf FlagGems
+          git clone https://github.com/flagos-ai/FlagGems.git
+          cd FlagGems
+          git checkout addflagtreetest
+          python3 -m pytest -s --mode=quick --limit-cases=1 --skipped-spectest

.github/workflows/hcu-build-and-test.yml

@@ -67,3 +67,17 @@ jobs:
           set -x
           cd third_party/hcu/python/test/unit
           sh flagtree_test.sh
+
+      - name: FlagTree Test with FlagGems
+        shell: bash
+        run: |
+          cd ../..
+          rm -rf FlagGems
+          git clone https://github.com/flagos-ai/FlagGems.git
+          cd FlagGems
+          git checkout addflagtreetest
+          python3 -m pytest -s --mode=quick --limit-cases=1 --skipped-spectest
+
+
+
+

.github/workflows/iluvatar-build-and-test.yml

@@ -68,3 +68,14 @@ jobs:
           python3 04-low-memory-dropout.py
           python3 07-extern-functions.py
           python3 08-grouped-gemm.py
+
+      - name: FlagTree Test with FlagGems
+        shell: bash
+        run: |
+          cd ../..
+          rm -rf FlagGems
+          git clone https://github.com/flagos-ai/FlagGems.git
+          cd FlagGems
+          git checkout addflagtreetest
+          python3 -m pytest -s --mode=quick --limit-cases=1 --skipped-spectest
+

.github/workflows/metax-build-and-test.yml

@@ -61,3 +61,14 @@ jobs:
         run: |
           set -x
           python3.10 -m pytest -s third_party/metax/python/test/unit
+
+      - name: FlagTree Test with FlagGems
+        shell: bash
+        run: |
+          cd ../..
+          rm -rf FlagGems
+          git clone https://github.com/flagos-ai/FlagGems.git
+          cd FlagGems
+          git checkout addflagtreetest
+          python3.10 -m pytest -s --mode=quick --limit-cases=1 --skipped-spectest
+

.github/workflows/mthreads-build-and-test.yml

@@ -68,3 +68,13 @@ jobs:
           ./python/build/cmake.linux-x86_64-cpython-3.10/bin/triton-opt --pass-pipeline='builtin.module(convert-triton-to-tritongpu{target="cuda:CC" num-warps=4 threads-per-warp=32 num-ctas=1})'  ./test/bin/mthreads/add_kernel.ttir
           ./python/build/cmake.linux-x86_64-cpython-3.10/bin/triton-opt --convert-ub-to-llvm ./test/bin/mthreads/add_kernel.ttgir
           MUSA_VISIBLE_DEVICES=7 python3 -m pytest -s third_party/mthreads/python/test/unit
+
+      - name: FlagTree Test with FlagGems
+        shell: bash
+        run: |
+          cd ../..
+          rm -rf FlagGems
+          git clone https://github.com/flagos-ai/FlagGems.git
+          cd FlagGems
+          git checkout addflagtreetest
+          python3 -m pytest -s --mode=quick --limit-cases=1 --skipped-spectest

.github/workflows/nv-build-and-test.yml

@@ -104,3 +104,14 @@ jobs:
           if [ -d "python/test/operators" ]; then
             python3.11 -m pytest -s python/test/operators
           fi
+
+      - name: FlagTree Test with FlagGems
+        shell: bash
+        run: |
+          cd ../..
+          rm -rf FlagGems
+          git clone https://github.com/flagos-ai/FlagGems.git
+          cd FlagGems
+          git checkout addflagtreetest
+          export GEMS_VENDOR=nvidia
+          python3.11 -m pytest -s --mode=quick --limit-cases=1 --skipped-spectest

.github/workflows/tsingmicro-build-and-test.yml

@@ -63,3 +63,13 @@ jobs:
           source ~/env.sh
           python3.11 -c 'import triton; print(triton.__path__)'
           /usr/local/lib/python3.11/dist-packages/triton/backends/tsingmicro/bin/tsingmicro-opt --version
+
+      - name: FlagTree Test with FlagGems
+        shell: bash
+        run: |
+          cd ../..
+          rm -rf FlagGems
+          git clone https://github.com/flagos-ai/FlagGems.git
+          cd FlagGems
+          git checkout addflagtreetest
+          python3.11 -m pytest -s --mode=quick --limit-cases=1 --skipped-spectest

.github/workflows/xpu-build-and-test.yml

@@ -67,3 +67,14 @@ jobs:
         run: |
           set -x
           python3 -m pytest -s third_party/xpu/python/test/unit
+
+      - name: FlagTree Test with FlagGems
+        shell: bash
+        run: |
+          cd ../..
+          rm -rf FlagGems
+          git clone https://github.com/flagos-ai/FlagGems.git
+          cd FlagGems
+          git checkout addflagtreetest
+          python3 -m pytest -s --mode=quick --limit-cases=1 --skipped-spectest
+