FlagTree/.github/workflows/tileir3.6-build-and-test.yml at main · flagos-ai/FlagTree · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
name: TileIR3.6-Build-And-Test

on:
  push:
    branches: [ "main", "triton_v3.6.x" ]
  pull_request:
    branches: [ "main", "triton_v3.6.x" ]

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

jobs:
  tileir36x-unit-test:
    runs-on: tileir3.6
    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: tileir

      - name: Build FlagTree
        if: steps.check_backend.outputs.should_skip != 'true'
        shell: bash
        run: |
          set -x
          source ~/env-3.6.sh
          /usr/local/cuda/bin/tileiras --version
          export FLAGTREE_BACKEND=tileir
          export CMAKE_BUILD_PARALLEL_LEVEL=32
          export TRITON_PTXAS_PATH="$PWD/third_party/nvidia/backend/bin/ptxas"
          export TRITON_PTXAS_BLACKWELL_PATH="$PWD/third_party/nvidia/backend/bin/ptxas-blackwell"
          MAX_JOBS=32 python3 -m pip install . --no-build-isolation

      - name: Unit Test
        if: steps.check_backend.outputs.should_skip != 'true'
        shell: bash
        run: |
          set -x
          source ~/env-3.6.sh
          rm -rf /tmp/flagtree_tileir_ci_01_cache /tmp/flagtree_tileir_ci_02_cache
          TRITON_CACHE_DIR=/tmp/flagtree_tileir_ci_01_cache \
            python3 python/tutorials/tileir/01-load-view-token-ordering.py
          TRITON_CACHE_DIR=/tmp/flagtree_tileir_ci_02_cache \
            python3 python/tutorials/tileir/02-mixed-kernel-routing.py
          python3 python/tutorials/tileir/03-triton-tileir-benchmarks.py \
            --case all \
            --ci-subset \
            --path both \
            --warmup 1 \
            --rep 1 \
            --min-rep 1 \
            --initial-rep 1 \
            --cache-dir /tmp/flagtree_tileir_ci_03_cache \
            --clear-cache