|
| 1 | +# SPDX-FileCopyrightText: 2026 Holochip Inc. |
| 2 | +# SPDX-License-Identifier: CC-BY-4.0 |
| 3 | +name: ML Inference CI |
| 4 | + |
| 5 | +on: |
| 6 | + push: |
| 7 | + paths: |
| 8 | + - 'attachments/ml_inference/**' |
| 9 | + - '.github/workflows/ml_inference_ci.yml' |
| 10 | + pull_request: |
| 11 | + paths: |
| 12 | + - 'attachments/ml_inference/**' |
| 13 | + - '.github/workflows/ml_inference_ci.yml' |
| 14 | + workflow_dispatch: |
| 15 | + |
| 16 | +jobs: |
| 17 | + desktop: |
| 18 | + name: Build (ubuntu-latest) |
| 19 | + runs-on: ubuntu-latest |
| 20 | + |
| 21 | + defaults: |
| 22 | + run: |
| 23 | + working-directory: attachments/ml_inference |
| 24 | + |
| 25 | + steps: |
| 26 | + - name: Checkout |
| 27 | + uses: actions/checkout@v4 |
| 28 | + |
| 29 | + - name: Install Clang + Ninja + ccache |
| 30 | + shell: bash |
| 31 | + run: | |
| 32 | + set -euo pipefail |
| 33 | + sudo apt-get update |
| 34 | + sudo apt-get install -y clang ninja-build ccache |
| 35 | +
|
| 36 | + - name: Select Clang toolchain |
| 37 | + shell: bash |
| 38 | + run: | |
| 39 | + echo "CC=clang" >> "$GITHUB_ENV" |
| 40 | + echo "CXX=clang++" >> "$GITHUB_ENV" |
| 41 | +
|
| 42 | + - name: ccache |
| 43 | + uses: actions/cache@v4 |
| 44 | + with: |
| 45 | + path: ~/.cache/ccache |
| 46 | + key: ${{ runner.os }}-ml-inference-ccache-${{ github.sha }} |
| 47 | + restore-keys: ${{ runner.os }}-ml-inference-ccache- |
| 48 | + |
| 49 | + - name: Cache Vulkan SDK |
| 50 | + id: cache-vulkan |
| 51 | + uses: actions/cache@v4 |
| 52 | + with: |
| 53 | + path: ${{ runner.temp }}/VulkanSDK |
| 54 | + key: ${{ runner.os }}-vulkan-sdk |
| 55 | + |
| 56 | + - name: Install Vulkan SDK |
| 57 | + shell: bash |
| 58 | + run: | |
| 59 | + set -euo pipefail |
| 60 | + sudo apt-get update |
| 61 | + sudo apt-get install -y curl ca-certificates xz-utils spirv-tools |
| 62 | +
|
| 63 | + SDK_DIR="${RUNNER_TEMP}/VulkanSDK" |
| 64 | +
|
| 65 | + if [ "${{ steps.cache-vulkan.outputs.cache-hit }}" != "true" ]; then |
| 66 | + echo "Downloading Vulkan SDK from LunarG..." |
| 67 | + SDK_TGZ="${RUNNER_TEMP}/vulkansdk-linux.tar.xz" |
| 68 | +
|
| 69 | + download_ok=0 |
| 70 | + for url in \ |
| 71 | + "https://sdk.lunarg.com/sdk/download/latest/linux/vulkan-sdk.tar.xz" \ |
| 72 | + "https://sdk.lunarg.com/sdk/download/latest/linux/vulkansdk-linux-x86_64.tar.xz" \ |
| 73 | + "https://sdk.lunarg.com/sdk/download/latest/linux/vulkan-sdk.tar.xz?Human=true" \ |
| 74 | + "https://sdk.lunarg.com/sdk/download/latest/linux/vulkansdk-linux-x86_64.tar.xz?Human=true" |
| 75 | + do |
| 76 | + echo "Attempting: $url" |
| 77 | + if curl -L --fail -o "$SDK_TGZ" "$url"; then |
| 78 | + download_ok=1 |
| 79 | + break |
| 80 | + fi |
| 81 | + done |
| 82 | + if [ "$download_ok" -ne 1 ]; then |
| 83 | + echo "Failed to download Vulkan SDK from LunarG (all endpoints returned non-200)." >&2 |
| 84 | + exit 1 |
| 85 | + fi |
| 86 | +
|
| 87 | + rm -rf "$SDK_DIR" |
| 88 | + mkdir -p "$SDK_DIR" |
| 89 | + tar -xJf "$SDK_TGZ" -C "$SDK_DIR" |
| 90 | + else |
| 91 | + echo "Using cached Vulkan SDK from $SDK_DIR" |
| 92 | + fi |
| 93 | +
|
| 94 | + VULKAN_SDK_PATH="$(find "$SDK_DIR" -maxdepth 1 -type d -name '1.*' | sort -r | head -n 1)" |
| 95 | + if [ -z "${VULKAN_SDK_PATH:-}" ]; then |
| 96 | + echo "Failed to locate extracted Vulkan SDK directory under $SDK_DIR" >&2 |
| 97 | + exit 1 |
| 98 | + fi |
| 99 | +
|
| 100 | + SDK_SYSROOT="$VULKAN_SDK_PATH" |
| 101 | + if [ -d "$VULKAN_SDK_PATH/x86_64" ]; then |
| 102 | + SDK_SYSROOT="$VULKAN_SDK_PATH/x86_64" |
| 103 | + fi |
| 104 | +
|
| 105 | + echo "VULKAN_SDK=$VULKAN_SDK_PATH" >> "$GITHUB_ENV" |
| 106 | + echo "VULKAN_SDK_SYSROOT=$SDK_SYSROOT" >> "$GITHUB_ENV" |
| 107 | +
|
| 108 | + if [ -d "$VULKAN_SDK_PATH/bin" ]; then |
| 109 | + echo "$VULKAN_SDK_PATH/bin" >> "$GITHUB_PATH" |
| 110 | + fi |
| 111 | + if [ -d "$SDK_SYSROOT/bin" ]; then |
| 112 | + echo "$SDK_SYSROOT/bin" >> "$GITHUB_PATH" |
| 113 | + fi |
| 114 | +
|
| 115 | + echo "CMAKE_PREFIX_PATH=$SDK_SYSROOT" >> "$GITHUB_ENV" |
| 116 | + echo "Vulkan_INCLUDE_DIR=$SDK_SYSROOT/include" >> "$GITHUB_ENV" |
| 117 | + if [ -f "$SDK_SYSROOT/lib/libvulkan.so" ]; then |
| 118 | + echo "Vulkan_LIBRARY=$SDK_SYSROOT/lib/libvulkan.so" >> "$GITHUB_ENV" |
| 119 | + elif [ -f "$SDK_SYSROOT/lib/libvulkan.so.1" ]; then |
| 120 | + echo "Vulkan_LIBRARY=$SDK_SYSROOT/lib/libvulkan.so.1" >> "$GITHUB_ENV" |
| 121 | + fi |
| 122 | +
|
| 123 | + if command -v slangc >/dev/null 2>&1; then |
| 124 | + echo "SLANGC_EXECUTABLE=$(command -v slangc)" >> "$GITHUB_ENV" |
| 125 | + fi |
| 126 | +
|
| 127 | + compat_dir="${RUNNER_TEMP}/slang-compat" |
| 128 | + mkdir -p "$compat_dir" |
| 129 | + pthread_path="$(ldconfig -p | awk '/libpthread\.so\.0/{print $NF; exit 0}' || true)" |
| 130 | + if [ -n "${pthread_path:-}" ] && [ -f "$pthread_path" ]; then |
| 131 | + ln -sf "$pthread_path" "$compat_dir/libpthread.so" |
| 132 | + fi |
| 133 | + echo "LD_LIBRARY_PATH=$compat_dir:${SDK_SYSROOT}/lib:${VULKAN_SDK_PATH}/lib:${LD_LIBRARY_PATH:-}" >> "$GITHUB_ENV" |
| 134 | +
|
| 135 | + - name: Vulkan SDK diagnostics |
| 136 | + shell: bash |
| 137 | + run: | |
| 138 | + echo "VULKAN_SDK=$VULKAN_SDK" |
| 139 | + echo "VULKAN_SDK_SYSROOT=${VULKAN_SDK_SYSROOT:-}" |
| 140 | + echo "Vulkan_INCLUDE_DIR=${Vulkan_INCLUDE_DIR:-}" |
| 141 | + echo "Vulkan_LIBRARY=${Vulkan_LIBRARY:-}" |
| 142 | + echo "SLANGC_EXECUTABLE=${SLANGC_EXECUTABLE:-}" |
| 143 | + echo "which slangc: $(command -v slangc || echo 'NOT FOUND')" |
| 144 | +
|
| 145 | + - name: Install system dependencies |
| 146 | + shell: bash |
| 147 | + run: | |
| 148 | + set -euo pipefail |
| 149 | + sudo apt-get update |
| 150 | + sudo apt-get install -y \ |
| 151 | + build-essential cmake git pkg-config \ |
| 152 | + ca-certificates curl zip unzip \ |
| 153 | + libglfw3-dev libglm-dev \ |
| 154 | + libx11-dev libxrandr-dev libxinerama-dev libxcursor-dev libxi-dev |
| 155 | +
|
| 156 | + - name: Clone NNEF parser |
| 157 | + shell: bash |
| 158 | + run: | |
| 159 | + set -euo pipefail |
| 160 | + NNEF_DIR="third_party/nnef-parser" |
| 161 | + if [ ! -f "${NNEF_DIR}/src/nnef.cpp" ]; then |
| 162 | + WORK="${RUNNER_TEMP}/NNEF-Tools" |
| 163 | + git clone --depth 1 https://github.com/KhronosGroup/NNEF-Tools.git "$WORK" |
| 164 | + mkdir -p "${NNEF_DIR}/src" "${NNEF_DIR}/include" |
| 165 | + cp "$WORK/nnef-pyproject/nnef/cpp/src/nnef.cpp" "${NNEF_DIR}/src/" |
| 166 | + cp -r "$WORK/nnef-pyproject/nnef/cpp/include/." "${NNEF_DIR}/include/" |
| 167 | + else |
| 168 | + echo "NNEF parser already present" |
| 169 | + fi |
| 170 | +
|
| 171 | + - name: Create placeholder resource directories |
| 172 | + shell: bash |
| 173 | + run: | |
| 174 | + # These directories are not tracked by git but are referenced in post-build |
| 175 | + # copy_directory commands. Create them so the build does not fail. |
| 176 | + mkdir -p models/mobilenetv2_nnef_optimized data sample_images |
| 177 | + touch mnist_model.onnx mnist_model.onnx.data mnist_weights.bin |
| 178 | +
|
| 179 | + - name: Configure |
| 180 | + shell: bash |
| 181 | + run: | |
| 182 | + set -euo pipefail |
| 183 | + extra_args=() |
| 184 | + if [ -n "${Vulkan_INCLUDE_DIR:-}" ]; then |
| 185 | + extra_args+=("-DVulkan_INCLUDE_DIR=${Vulkan_INCLUDE_DIR}") |
| 186 | + fi |
| 187 | + if [ -n "${Vulkan_LIBRARY:-}" ]; then |
| 188 | + extra_args+=("-DVulkan_LIBRARY=${Vulkan_LIBRARY}") |
| 189 | + fi |
| 190 | + if [ -n "${SLANGC_EXECUTABLE:-}" ]; then |
| 191 | + extra_args+=("-DSLANGC_EXECUTABLE=${SLANGC_EXECUTABLE}") |
| 192 | + fi |
| 193 | +
|
| 194 | + cmake -S . -B build -G Ninja \ |
| 195 | + -DCMAKE_BUILD_TYPE=Release \ |
| 196 | + -DCMAKE_C_COMPILER=clang \ |
| 197 | + -DCMAKE_CXX_COMPILER=clang++ \ |
| 198 | + -DCMAKE_CXX_COMPILER_LAUNCHER=ccache \ |
| 199 | + -DCMAKE_C_COMPILER_LAUNCHER=ccache \ |
| 200 | + -DCMAKE_PREFIX_PATH="${VULKAN_SDK_SYSROOT}" \ |
| 201 | + -DML_SETUP_PYTHON_ENV=OFF \ |
| 202 | + -DML_TRAIN_MODELS=OFF \ |
| 203 | + "${extra_args[@]}" |
| 204 | +
|
| 205 | + - name: Build |
| 206 | + shell: bash |
| 207 | + run: cmake --build build --parallel 4 |
| 208 | + |
| 209 | + - name: ccache stats |
| 210 | + shell: bash |
| 211 | + run: ccache -s |
0 commit comments