diff --git a/.github/workflows/aipu-build-and-test.yml b/.github/workflows/aipu-build-and-test.yml index 23f04957c..beac4beb7 100644 --- a/.github/workflows/aipu-build-and-test.yml +++ b/.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 diff --git a/.github/workflows/ascend-build-and-test.yml b/.github/workflows/ascend-build-and-test.yml index 3ba1d4f64..feda16fd3 100644 --- a/.github/workflows/ascend-build-and-test.yml +++ b/.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 \ No newline at end of file diff --git a/.github/workflows/cambricon-build-and-test.yml b/.github/workflows/cambricon-build-and-test.yml index e8b163d2a..643643c5d 100644 --- a/.github/workflows/cambricon-build-and-test.yml +++ b/.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 diff --git a/.github/workflows/hcu-build-and-test.yml b/.github/workflows/hcu-build-and-test.yml index 95da538a8..af54216be 100644 --- a/.github/workflows/hcu-build-and-test.yml +++ b/.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 + + + + diff --git a/.github/workflows/iluvatar-build-and-test.yml b/.github/workflows/iluvatar-build-and-test.yml index f5d30333e..7a1923939 100644 --- a/.github/workflows/iluvatar-build-and-test.yml +++ b/.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 + diff --git a/.github/workflows/metax-build-and-test.yml b/.github/workflows/metax-build-and-test.yml index 4f7a2252b..fe5802dda 100644 --- a/.github/workflows/metax-build-and-test.yml +++ b/.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 + diff --git a/.github/workflows/mthreads-build-and-test.yml b/.github/workflows/mthreads-build-and-test.yml index 33182c125..0842495e6 100644 --- a/.github/workflows/mthreads-build-and-test.yml +++ b/.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 diff --git a/.github/workflows/nv-build-and-test.yml b/.github/workflows/nv-build-and-test.yml index cddaddc44..efbdb0020 100644 --- a/.github/workflows/nv-build-and-test.yml +++ b/.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 diff --git a/.github/workflows/tsingmicro-build-and-test.yml b/.github/workflows/tsingmicro-build-and-test.yml index b64a3d3de..fb954f0d8 100644 --- a/.github/workflows/tsingmicro-build-and-test.yml +++ b/.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 diff --git a/.github/workflows/xpu-build-and-test.yml b/.github/workflows/xpu-build-and-test.yml index 32bbfabdc..5a8869dbd 100644 --- a/.github/workflows/xpu-build-and-test.yml +++ b/.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 +