Skip to content

Commit 012d762

Browse files
committed
ci: use devws HRX scripts
1 parent 9fc0cdb commit 012d762

1 file changed

Lines changed: 52 additions & 14 deletions

File tree

.github/workflows/build-hrx.yml

Lines changed: 52 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,11 @@ jobs:
2626
matrix:
2727
include:
2828
- name: gfx1151_strix-halo
29+
gpu_target: gfx1151
2930
runner: linux-gfx1151-gpu-rocm
3031
rocm_tarball: therock-dist-linux-gfx1151-7.14.0a20260617.tar.gz
3132
- name: gfx1201_9070
33+
gpu_target: gfx1201
3234
runner: linux-gfx120X-gpu-rocm
3335
rocm_tarball: therock-dist-linux-gfx120X-all-7.14.0a20260617.tar.gz
3436
runs-on: ${{ matrix.runner }}
@@ -44,7 +46,7 @@ jobs:
4446
--device /dev/dri
4547
env:
4648
DEVWS_REPOSITORY: AaronStGeorge/llamacpp-devws
47-
DEVWS_REF: build-hrx-v2
49+
DEVWS_REF: ci-scripts
4850
DEVWS_DIR: /work/llamacpp-devws
4951
ROCM_DIR: /work/rocm
5052
ROCM_TARBALL_BASE_URL: https://rocm.nightlies.amd.com/tarball-multi-arch
@@ -74,14 +76,11 @@ jobs:
7476

7577
- name: Runner identity
7678
if: always()
77-
run: |
78-
printf 'RUNNER_NAME=%s\n' "${RUNNER_NAME:-}"
79-
printf 'RUNNER_OS=%s\n' "${RUNNER_OS:-}"
80-
printf 'GITHUB_EVENT_NAME=%s\n' "${GITHUB_EVENT_NAME:-}"
81-
hostname
82-
uname -a
83-
ls -l /dev/kfd /dev/dri || true
84-
printenv | sort | grep -E '^(GITHUB_|RUNNER_|ROCM_|HSA_|HIP_)' || true
79+
env:
80+
MATRIX_NAME: ${{ matrix.name }}
81+
MATRIX_GPU_TARGET: ${{ matrix.gpu_target }}
82+
MATRIX_RUNS_ON: ${{ matrix.runner }}
83+
run: "${GITHUB_WORKSPACE}/devws-src/ci/runner-info.sh"
8584

8685
- name: Install host build dependencies
8786
run: |
@@ -104,12 +103,10 @@ jobs:
104103
- name: Stage dev workspace under /work
105104
run: |
106105
rm -rf "${DEVWS_DIR}"
107-
mkdir -p "${DEVWS_DIR}/sources"
106+
mkdir -p "${DEVWS_DIR}"
108107
cp -a "${GITHUB_WORKSPACE}/devws-src/." "${DEVWS_DIR}/"
109-
cp -a "${GITHUB_WORKSPACE}/llama-src" "${DEVWS_DIR}/sources/llama.cpp"
110-
cp -a "${GITHUB_WORKSPACE}/hrx-system-src" "${DEVWS_DIR}/sources/hrx-system"
111-
git -C "${DEVWS_DIR}/sources/llama.cpp" checkout -B hrx-v2
112-
git -C "${DEVWS_DIR}/sources/hrx-system" checkout -B main
108+
git -C "${GITHUB_WORKSPACE}/llama-src" checkout -B hrx-v2
109+
git -C "${GITHUB_WORKSPACE}/hrx-system-src" checkout -B main
113110
114111
- name: Fetch ROCm nightly tarball
115112
run: |
@@ -138,9 +135,50 @@ jobs:
138135
--action loom \
139136
--action rocm-health \
140137
--action llama-hrx2 \
138+
--hrx-source "${GITHUB_WORKSPACE}/hrx-system-src" \
139+
--llama-source "${GITHUB_WORKSPACE}/llama-src" \
141140
--gfx-targets auto \
142141
--jobs "$(nproc)"
143142
144143
- name: Verify HRX2 build configuration
145144
run: |
146145
grep -F 'GGML_HRX2:BOOL=ON' "${DEVWS_DIR}/build/llama-hrx2/CMakeCache.txt"
146+
147+
- name: Run sample MUL_MAT correctness config on HRX2
148+
run: |
149+
export ROCM_PATH="${DEVWS_DIR}/rocm"
150+
export GGML_HRX_ROCM_PATH="${DEVWS_DIR}/rocm"
151+
export PATH="${DEVWS_DIR}/rocm/bin:${DEVWS_DIR}/rocm/lib/llvm/bin:${PATH}"
152+
export LD_LIBRARY_PATH="${DEVWS_DIR}/build/llama-hrx2/bin:${DEVWS_DIR}/build/hrx-install/lib:${DEVWS_DIR}/build/hrx-install/lib64:${DEVWS_DIR}/rocm/lib:${DEVWS_DIR}/rocm/lib64:${DEVWS_DIR}/rocm/lib/rocm_sysdeps/lib:${LD_LIBRARY_PATH:-}"
153+
export IREE_HAL_AMDGPU_LIBHSA_PATH="${DEVWS_DIR}/rocm/lib/libhsa-runtime64.so"
154+
mkdir -p "${GITHUB_WORKSPACE}/benchmark-results"
155+
"${DEVWS_DIR}/ci/tools/run-op-test.py" \
156+
--test-backend-ops "${DEVWS_DIR}/build/llama-hrx2/bin/test-backend-ops" \
157+
--test-file "${DEVWS_DIR}/ci/benchmark-configs/test/mul_mat_f16.txt" \
158+
--op MUL_MAT \
159+
--backend HRX20 \
160+
--output "${GITHUB_WORKSPACE}/benchmark-results/sample-mul-mat-f16-hrx2-test.jsonl" \
161+
--raw-output "${GITHUB_WORKSPACE}/benchmark-results/sample-mul-mat-f16-hrx2-test.txt"
162+
163+
- name: Run sample MUL_MAT benchmark config on HRX2
164+
run: |
165+
export ROCM_PATH="${DEVWS_DIR}/rocm"
166+
export GGML_HRX_ROCM_PATH="${DEVWS_DIR}/rocm"
167+
export PATH="${DEVWS_DIR}/rocm/bin:${DEVWS_DIR}/rocm/lib/llvm/bin:${PATH}"
168+
export LD_LIBRARY_PATH="${DEVWS_DIR}/build/llama-hrx2/bin:${DEVWS_DIR}/build/hrx-install/lib:${DEVWS_DIR}/build/hrx-install/lib64:${DEVWS_DIR}/rocm/lib:${DEVWS_DIR}/rocm/lib64:${DEVWS_DIR}/rocm/lib/rocm_sysdeps/lib:${LD_LIBRARY_PATH:-}"
169+
export IREE_HAL_AMDGPU_LIBHSA_PATH="${DEVWS_DIR}/rocm/lib/libhsa-runtime64.so"
170+
mkdir -p "${GITHUB_WORKSPACE}/benchmark-results"
171+
"${DEVWS_DIR}/ci/tools/run-op-perf.py" \
172+
--test-backend-ops "${DEVWS_DIR}/build/llama-hrx2/bin/test-backend-ops" \
173+
--test-file "${DEVWS_DIR}/ci/benchmark-configs/test/mul_mat_f16.txt" \
174+
--op MUL_MAT \
175+
--backend HRX20 \
176+
--output "${GITHUB_WORKSPACE}/benchmark-results/sample-mul-mat-f16-hrx2-perf.jsonl" \
177+
--raw-output "${GITHUB_WORKSPACE}/benchmark-results/sample-mul-mat-f16-hrx2-perf.txt"
178+
179+
- name: Upload HRX2 benchmark results
180+
uses: actions/upload-artifact@v5
181+
with:
182+
name: hrx2-benchmark-results-${{ matrix.name }}
183+
path: benchmark-results/
184+
if-no-files-found: error

0 commit comments

Comments
 (0)