amd3.6-build-and-test.yml

# Copyright 2025-     FlagOS Contributors
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

name: Amd3.6-Build-And-Test

on:
  push:
    branches: [ "main", "triton_v3.6.x" ]
  pull_request:
    branches: [ "main", "triton_v3.6.x" ]
    types: [opened, synchronize, reopened, ready_for_review]

permissions:
  contents: read
  pull-requests: read

concurrency:
  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
  cancel-in-progress: true

jobs:
  amd36x-unit-test:
    runs-on: amd3.6
    if: ${{ github.repository == 'flagos-ai/flagtree' && github.event.pull_request.draft == false }}
    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: amd

      - name: Cleanup FlagTree Environment
        if: ${{ steps.check_backend.outputs.should_skip != 'true' }}
        uses: flagos-ai/FlagTree/.github/actions/cleanup-flagtree-env@main

      - name: Detect Target Branch
        shell: bash
        run: |
          set -x
          if [ "${{ github.event_name }}" = "pull_request" ]; then
            TARGET_BRANCH="${{ github.base_ref }}"
          else
            TARGET_BRANCH="${{ github.ref_name }}"
          fi
          echo "TARGET_BRANCH=$TARGET_BRANCH" >> $GITHUB_ENV
          echo "TARGET_BRANCH=$TARGET_BRANCH"

      - name: Build FlagTree
        if: ${{ steps.check_backend.outputs.should_skip != 'true' }}
        shell: bash
        run: |
          set -x
          source ~/env-3.6.sh
          MAX_JOBS=32 python3 -m pip install . --no-build-isolation -v

      - name: Clear cache if ClearCache label present
        if: steps.check_backend.outputs.should_skip != 'true'
        id: clear_cache
        uses: flagos-ai/FlagTree/.github/actions/clear-cache-if-labeled@main
        with:
          github-token: ${{ secrets.GITHUB_TOKEN }}
          backend: amd

      - name: Unit Test
        if: ${{ steps.check_backend.outputs.should_skip != 'true' }}
        shell: bash
        run: |
          set -x
          source ~/env.sh
          ## python tutorials
          python3 python/tutorials/01-vector-add.py --only_unit_test
          python3 python/tutorials/02-fused-softmax.py --only_unit_test
          python3 python/tutorials/03-matrix-multiplication.py --only_unit_test
          python3 python/tutorials/04-low-memory-dropout.py --only_unit_test
          python3 python/tutorials/05-layer-norm.py --only_unit_test
          # python3 python/tutorials/06-fused-attention.py --only_unit_test
          python3 python/tutorials/07-extern-functions.py --only_unit_test
          python3 python/tutorials/08-grouped-gemm.py --only_unit_test
          # python3 python/tutorials/09-persistent-matmul.py --only_unit_test
          # python3 python/tutorials/11-programmatic-dependent-launch.py --only_unit_test
          # python3 python/tutorials/12-fused_inv_rope_fp8_quant_reorder.py --only_unit_test
          ## python unit test
          python3 -m pytest -s python/test/unit/cuda
          python3 -m pytest -s python/test/unit/instrumentation
          python3 -m pytest -s python/test/unit/language \
              -k "not test_typeconvert_upcast and \
                  not test_typeconvert_downcast" \
              --ignore=python/test/unit/language/test_core.py \
              --ignore=python/test/unit/language/test_line_info.py
          python3 -m pytest -s --first-parameter-only \
              python/test/unit/language/test_core.py \
              -k "not test_sum_dtype and \
                  not test_dot_multidim"
          python3 -m pytest -s python/test/unit/runtime
          if [ -d "python/test/operators" ]; then
            python3 -m pytest -s python/test/operators
          fi
          ## flagtree tle python tutorials
          python3 python/tutorials/tle/01-fft.py
          # python3 python/tutorials/tle/02-moe_align_block_size.py
          python3 python/tutorials/tle/03-topk.py
          # python3 python/tutorials/tle/04-cluster-gemm.py
          # python3 python/tutorials/tle/deepseek_v32/01-topk_selector.py
          # python3 python/tutorials/tle/deepseek_v32/02-sparse-mla.py
          ## flagtree tle python unit test
          python3 -m pytest -s python/test/tle/integration \
              --ignore=python/test/tle/integration/test_tle_distributed.py \
              --ignore=python/test/tle/integration/test_tle_tma_copy.py
          python3 -m pytest -s python/test/tle/unit \
              --ignore=python/test/tle/unit/test_tle_distributed_d2d.py \
              --ignore=python/test/tle/unit/test_tle_get_local_pe.py \
              --ignore=python/test/tle/unit/test_tle_d2d_barrier.py \
              --ignore=python/test/tle/unit/test_tle_get_node_rank.py