IRON/.github/workflows/strix-examples.yml at 1bd0b20d051c9be89803bc8629d9dca0a17fcb99 · amd/IRON · 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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
# SPDX-FileCopyrightText: Copyright (C) 2025 Advanced Micro Devices, Inc. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

name: Test Example Applications

on:
  workflow_dispatch:
  pull_request:
  push:
    branches:
      - main
      - devel

concurrency:
  group: ${{ github.workflow }}-${{ github.ref }}
  cancel-in-progress: true

jobs:
  Test:
    runs-on: [self-hosted, strix]
    steps:
      - name: Checkout
        uses: actions/checkout@v4
        env:
          HOME: /workspace

      - name: Prerequisites
        uses: ./.github/actions/prereqs
        with:
          env_name: ci_env

      - name: Install additional requirements for examples
        shell: bash
        env:
          HOME: /workspace
        run: |
          set -euxo pipefail
          source ci_env/bin/activate
          pip install -r requirements_examples.txt

      - name: Print mlir-aie version
        shell: bash
        env:
          HOME: /workspace
        run: |
          set -euxo pipefail
          source ci_env/bin/activate
          echo "=== mlir-aie version ==="
          aie-opt --version
          echo "========================"

      - name: Print resource limits
        shell: bash
        run: |
          echo "=== ulimit -a ==="
          ulimit -a
          echo ""
          echo "=== /proc/self/limits ==="
          cat /proc/self/limits
          echo ""
          echo "=== Free memory ==="
          free -h

      - name: Run examples
        id: test
        continue-on-error: true
        uses: ./.github/actions/test
        env:
          PYTHONUNBUFFERED: "1"
        with:
          env_name: ci_env
          test_suite: "applications"

      - name: Commit example test results
        uses: ./.github/actions/commit_results
        id: commit_results
        with:
          dir: strix/examples
          pretty_flags: "--metric='TTFT (mean)' --metric='TPS (mean)' -o pretty.md"

      - name: Upload test results
        uses: actions/upload-artifact@v4
        with:
          name: results-examples
          path: ${{ steps.commit_results.outputs.results_dir }}
          retention-days: 14

      - name: Upload MLIR artifacts for debugging
        if: always()
        uses: actions/upload-artifact@v4
        with:
          name: mlir-artifacts
          path: |
            iron/applications/llama_3.2_1b/build/*.mlir
            iron/applications/llama_3.2_1b/build_elf/*.mlir
          retention-days: 14
          if-no-files-found: warn

      - name: Fail workflow if examples failed
        if: steps.test.outcome == 'failure'
        run: exit 1