[CI] Update build and test files on triton_v3.3.x branch to use check… #342
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: AIPU-Build-And-Test | |
| on: | |
| push: | |
| branches: [ "triton_v3.3.x" ] | |
| pull_request: | |
| branches: [ "triton_v3.3.x" ] | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }} | |
| cancel-in-progress: true | |
| jobs: | |
| aipu-build-and-test: | |
| runs-on: aipu | |
| if: ${{ github.repository == 'FlagTree/flagtree' || github.repository == 'flagos-ai/flagtree' }} | |
| steps: | |
| - name: Setup environment | |
| shell: bash | |
| run: | | |
| source ~/env.sh | |
| env | grep -E '^(http_proxy|https_proxy|all_proxy|no_proxy)=' >> $GITHUB_ENV || true | |
| - name: Smart Checkout | |
| uses: flagos-ai/FlagTree/.github/actions/smart-checkout@main | |
| with: | |
| checkout_version: 'v6' | |
| - name: Check if backend-relevant files changed | |
| id: check_backend | |
| uses: flagos-ai/FlagTree/.github/actions/check-backend-changed@main | |
| with: | |
| backend: aipu | |
| - name: FlagTree Build on AIPU | |
| if: steps.check_backend.outputs.should_skip != 'true' | |
| shell: bash | |
| run: | | |
| set -x | |
| pip uninstall -y triton | |
| source ~/env_setup.sh | |
| export FLAGTREE_BACKEND=aipu | |
| cd python | |
| MAX_JOBS=32 python3.10 -m pip install . --no-build-isolation | |
| - name: FlagTree Test on AIPU | |
| if: steps.check_backend.outputs.should_skip != 'true' | |
| shell: bash | |
| run: | | |
| set -x | |
| source ~/env_setup.sh | |
| python3.10 third_party/aipu/python/test/test_01_vector_add.py | |
| python3.10 third_party/aipu/python/test/test_01_vector_add_shared_memory.py | |
| python3.10 third_party/aipu/python/test/test_01_vector_add_dma.py | |
| python3.10 third_party/aipu/python/test/test_02_fused_softmax.py | |
| python3.10 third_party/aipu/python/test/test_02_fused_softmax_shared_memory.py | |
| python3.10 third_party/aipu/python/test/test_02_fused_softmax_dma.py | |
| python3.10 third_party/aipu/python/test/test_libdevice_fmod.py | |
| python3.10 third_party/aipu/python/test/test_libdevice_fmod_shared_memory.py | |
| python3.10 third_party/aipu/python/test/test_libdevice_fmod_dma.py | |
| python3.10 third_party/aipu/python/test/test_libdevice_pow.py | |
| python3.10 third_party/aipu/python/test/test_libdevice_pow_shared_memory.py | |
| python3.10 third_party/aipu/python/test/test_libdevice_pow_dma.py | |
| python3.10 third_party/aipu/python/test/test_codegen_dma.py | |
| python3.10 third_party/aipu/python/test/test_01_vector_add_structured_1d.py | |
| python3.10 third_party/aipu/python/test/test_01_vector_add_structured_1d_shared_memory.py | |
| python3.10 third_party/aipu/python/test/test_01_vector_add_structured_1d_dma.py | |
| python3.10 third_party/aipu/python/test/min.py | |
| python3.10 third_party/aipu/python/test/min_shared_memory.py | |
| #python3.10 third_party/aipu/python/test/min_dma.py | |
| 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 |