(T3K) T3000 unit tests #1573
This file contains 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: "(T3K) T3000 unit tests" | |
| on: | |
| workflow_dispatch: | |
| schedule: | |
| - cron: "0 */3 * * *" # This cron schedule runs the workflow every 3 hours | |
| jobs: | |
| build-artifact: | |
| uses: ./.github/workflows/build-artifact.yaml | |
| with: | |
| arch: '["wormhole_b0"]' | |
| secrets: inherit | |
| t3000-unit-tests: | |
| needs: build-artifact | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| test-group: [ | |
| { name: "t3k ttmetal tests", arch: wormhole_b0, cmd: run_t3000_ttmetal_tests, timeout: 30, owner_id: ULMEPM2MA}, #Sean Nijjar | |
| { name: "t3k ttnn tests", arch: wormhole_b0, cmd: run_t3000_ttnn_tests, timeout: 120, owner_id: UBHPP2NDP}, #Joseph Chu | |
| { name: "t3k falcon7b tests", arch: wormhole_b0, cmd: run_t3000_falcon7b_tests, timeout: 30, owner_id: UBHPP2NDP}, #Joseph Chu | |
| { name: "t3k falcon40b tests", arch: wormhole_b0, cmd: run_t3000_falcon40b_tests, timeout: 30, owner_id: U053W15B6JF}, #Djordje Ivanovic | |
| { name: "t3k mixtral tests", arch: wormhole_b0, cmd: run_t3000_mixtral_tests, timeout: 30, owner_id: U03PUAKE719}, #Miguel Tairum Cruz | |
| { name: "t3k grok tests", arch: wormhole_b0, cmd: run_t3000_grok_tests, timeout: 30, owner_id: U03HY7MK4BT}, #Mark O'Connor | |
| { name: "t3k unet shallow tests", arch: wormhole_b0, cmd: run_t3000_unet_shallow_tests, timeout: 30, owner_id: U06ECNVR0EN}, #Evan Smal | |
| ] | |
| name: ${{ matrix.test-group.name }} | |
| env: | |
| TT_METAL_ENV: ${{ vars.TT_METAL_ENV }} | |
| ARCH_NAME: ${{ matrix.test-group.arch }} | |
| LOGURU_LEVEL: INFO | |
| LD_LIBRARY_PATH: ${{ github.workspace }}/build/lib | |
| environment: dev | |
| runs-on: ["arch-wormhole_b0", "config-t3000", "in-service", "pipeline-functional"] | |
| steps: | |
| - uses: tenstorrent-metal/metal-workflows/.github/actions/[email protected] | |
| - name: Set up dynamic env vars for build | |
| run: | | |
| echo "TT_METAL_HOME=$(pwd)" >> $GITHUB_ENV | |
| - uses: actions/download-artifact@v4 | |
| with: | |
| name: TTMetal_build_${{ matrix.test-group.arch }} | |
| - name: Extract files | |
| run: tar -xvf ttm_${{ matrix.test-group.arch }}.tar | |
| - uses: ./.github/actions/install-python-deps | |
| - name: Run unit regression tests | |
| shell: bash {0} | |
| timeout-minutes: ${{ matrix.test-group.timeout }} | |
| run: | | |
| source ${{ github.workspace }}/python_env/bin/activate | |
| cd $TT_METAL_HOME | |
| export PYTHONPATH=$TT_METAL_HOME | |
| source ${{ github.workspace }}/tests/scripts/t3000/run_t3000_unit_tests.sh | |
| ${{ matrix.test-group.cmd }} | |
| - uses: ./.github/actions/slack-report | |
| if: ${{ failure() }} | |
| with: | |
| slack_webhook_url: ${{ secrets.SLACK_WEBHOOK_URL }} | |
| owner: ${{ matrix.test-group.owner_id }} |