[Feat] Add GQA/MHA FlashAttention-3 baseline #41
Workflow file for this run
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: hello-self-hosted | |
| on: | |
| push: | |
| branches: ['**'] # All branches | |
| tags: ['**'] # All tags push will also trigger, can be removed as needed | |
| pull_request: | |
| branches: ['**'] # All branch PRs trigger CI | |
| jobs: | |
| pre-commit: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Checkout code | |
| uses: actions/checkout@v3 | |
| with: | |
| fetch-depth: 0 # Equivalent to GIT_STRATEGY: fetch | |
| - name: Set up Python | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: "3.10" # At least 3.9 to avoid pre-commit-hooks reporting Python version too low | |
| - name: Run pre-commit | |
| # Official pre-commit Action, automatically: | |
| # 1) Install pre-commit | |
| # 2) Generate virtual environment according to .pre-commit-config.yaml configuration | |
| # 3) Execute pre-commit run | |
| uses: pre-commit/[email protected] | |
| with: | |
| # Run on all files once, and print diff on failure | |
| extra_args: --all-files --show-diff-on-failure | |
| tileops_test_0-1-6-post1: | |
| needs: pre-commit | |
| runs-on: [self-hosted, tile-ops] | |
| steps: | |
| - name: Checkout code | |
| uses: actions/checkout@v3 | |
| with: | |
| fetch-depth: 0 # Equivalent to GIT_STRATEGY: fetch | |
| - name: Setup & Run tests | |
| run: | | |
| source ~/.bashrc | |
| export PATH=/usr/local/cuda-12.9/bin:$PATH | |
| export LD_LIBRARY_PATH=/usr/local/cuda-12.9/lib64:$LD_LIBRARY_PATH | |
| source /home/lyc/miniconda3/etc/profile.d/conda.sh | |
| conda activate tileops-release | |
| export PYTHONPATH="$(pwd):$PYTHONPATH" | |
| echo "PYTHONPATH=$PYTHONPATH" | |
| bash ci-test.sh tileops_test_0_1_6.log | |
| shell: bash | |
| - name: Upload artifacts | |
| uses: actions/upload-artifact@v4 | |
| if: always() # Equivalent to when: always | |
| with: | |
| name: tileops_test_0_1_6.log | |
| path: tileops_test_0_1_6.log | |
| retention-days: 7 # Equivalent to expire_in: 1 week | |
| tileops_test_nightly: | |
| needs: pre-commit | |
| runs-on: [self-hosted, tile-ops] | |
| steps: | |
| - name: Checkout code | |
| uses: actions/checkout@v3 | |
| with: | |
| fetch-depth: 0 # Equivalent to GIT_STRATEGY: fetch | |
| - name: Setup & Run tests | |
| run: | | |
| source ~/.bashrc | |
| export PATH=/usr/local/cuda-12.9/bin:$PATH | |
| export LD_LIBRARY_PATH=/usr/local/cuda-12.9/lib64:$LD_LIBRARY_PATH | |
| source /home/lyc/miniconda3/etc/profile.d/conda.sh | |
| conda activate tileops-nightly | |
| export PYTHONPATH="$(pwd):$PYTHONPATH" | |
| echo "PYTHONPATH=$PYTHONPATH" | |
| bash ci-test.sh tileops_test_nightly.log | |
| shell: bash | |
| - name: Upload artifacts | |
| uses: actions/upload-artifact@v4 | |
| if: always() # Equivalent to when: always | |
| with: | |
| name: tileops_test_nightly.log | |
| path: tileops_test_nightly.log | |
| retention-days: 7 # Equivalent to expire_in: 1 week | |
| tileops_profile_nightly: | |
| needs: [pre-commit, tileops_test_nightly] | |
| runs-on: [self-hosted, tile-ops] | |
| steps: | |
| - name: Checkout code | |
| uses: actions/checkout@v3 | |
| with: | |
| fetch-depth: 0 # Equivalent to GIT_STRATEGY: fetch | |
| - name: Setup & Run tests | |
| run: | | |
| source ~/.bashrc | |
| export PATH=/usr/local/cuda-12.9/bin:$PATH | |
| export LD_LIBRARY_PATH=/usr/local/cuda-12.9/lib64:$LD_LIBRARY_PATH | |
| source /home/lyc/miniconda3/etc/profile.d/conda.sh | |
| conda activate tileops-nightly | |
| export PYTHONPATH="$(pwd):$PYTHONPATH" | |
| echo "PYTHONPATH=$PYTHONPATH" | |
| bash profile-run.sh --log profile_out/tileops_profile_nightly.log | |
| shell: bash | |
| - name: Upload profile_out artifacts | |
| uses: actions/upload-artifact@v4 | |
| if: always() | |
| with: | |
| name: profile_out | |
| path: profile_out/ | |
| retention-days: 7 |