Skip to content

Commit 53e1b2f

Browse files
irvineoyirvineoy
committed
Upgrade MI355X runtime and evaluation tools to ROCm 10
Pin the new scoring image, adapt SDK loading and task dependencies, and update all six evaluation-tool runtimes with matching documentation and regression coverage. Keep qualification pending for GPU ASan and the existing fused-MoE numerical failures.
1 parent be6ad00 commit 53e1b2f

109 files changed

Lines changed: 6973 additions & 271 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎README.md‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ The prompt system also recognizes `cuda2hip`; the current bundled task tree does
167167
- Node.js 22+ and npm when using the alternative npm installation of Claude Code
168168
(or another npm-installed agent CLI); DeepSeek Harness uses a dedicated
169169
Node.js 24 prefix as described in [its setup guide](agents/deepseek_harness/README.md)
170-
- For MI300/MI355X, use the GPU-specific SGLang image: `gfx942` uses `lmsysorg/sglang:v0.5.12-rocm720-mi30x`; `gfx950` uses `lmsysorg/sglang-rocm:v0.5.14-rocm720-mi35x-20260705`
170+
- For MI300/MI355X, use the GPU-specific SGLang image: `gfx942` uses `lmsysorg/sglang:v0.5.12-rocm720-mi30x`; `gfx950` uses the digest-pinned SGLang 0.5.20 / ROCm 10 runtime. See the [runtime compatibility matrix](docs/reference/compatibility-matrix.md) for image selection and task-specific requirements.
171171
- For RDNA4 `gfx1201`, the runner automatically builds the default [pinned RDNA4 runtime](docker/rdna4/README.md) on first use if it is missing.
172172
- A supported agent CLI installed and logged in on the host, or the dependencies required by a specialized agent
173173

‎agents/geak/README.md‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,11 @@ Docker provisions Claude for `geak` and its v2 aliases, mounts only the selected
6363
GEAK checkout read-only, and forwards `GEAK_HOME`. The complete checkout is
6464
needed for revision checks and private engine/knowledge copies. Preflight
6565
installs the pinned SDK into the GEAK-only dependency directory when necessary
66-
and verifies the clean upstream pin. These checks do not certify a live
66+
and verifies the clean upstream pin. SDK dependencies are cached under
67+
`.aka-pyuserbase/geak-sdk/<python-abi>` so changing the image's Python version
68+
installs compatible wheels. The first run after an upgrade may reinstall the
69+
SDK; existing caches remain available for their original interpreter.
70+
These checks do not certify a live
6771
Workflow invocation or a GPU task.
6872

6973
## Upstream compatibility and evaluation

‎docker/eval-tools/gpu-asan/Dockerfile‎

Lines changed: 35 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1,30 +1,48 @@
1-
ARG BASE_IMAGE=lmsysorg/sglang-rocm:v0.5.14-rocm720-mi35x-20260705@sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78
1+
ARG BASE_IMAGE=lmsysorg/sglang:v0.5.20-rocm10-mi35x@sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69
22
FROM ${BASE_IMAGE}
33

44
LABEL org.opencontainers.image.title="AgentKernelArena ROCm GPU ASan tool runtime" \
5-
org.opencontainers.image.base.digest="sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78" \
6-
org.opencontainers.image.version="rocm-7.2.0-asan"
5+
org.opencontainers.image.base.digest="sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69" \
6+
org.opencontainers.image.version="rocm-10.0.0-asan"
77

88
COPY docker/eval-tools/gpu-asan/packages.sha256 /tmp/packages.sha256
99

10-
RUN mkdir -p /tmp/gpu-asan-debs \
10+
# Unpack checksum-locked ROCm 10 ASan packages into their own versioned prefix.
11+
# Installing their metapackage would also mutate the base image's SDK selection.
12+
RUN mkdir -p /tmp/gpu-asan-debs /tmp/gpu-asan-root \
1113
&& cd /tmp/gpu-asan-debs \
12-
&& apt-get update \
13-
&& apt-get download \
14-
rocm-core-asan=7.2.0.70200-43~22.04 \
15-
comgr-asan=3.0.0.70200-43~22.04 \
16-
hsa-rocr-asan=1.18.0.70200-43~22.04 \
17-
hip-runtime-amd-asan=7.2.26015.70200-43~22.04 \
14+
&& while read -r checksum package; do \
15+
curl --fail --location --retry 3 -o "$package" \
16+
"https://stable.repo.amd.com/rocm/core/packages-asan/ubuntu2404/pool/main/$package"; \
17+
done < /tmp/packages.sha256 \
1818
&& sha256sum --check /tmp/packages.sha256 \
19-
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ./*.deb \
20-
&& rm -rf /tmp/gpu-asan-debs /tmp/packages.sha256 /var/lib/apt/lists/*
19+
&& for package in *.deb; do dpkg-deb --extract "$package" /tmp/gpu-asan-root; done \
20+
&& mv /tmp/gpu-asan-root/opt/rocm/core-asan-10.0 /opt/rocm-asan-10.0 \
21+
&& test -L /opt/rocm \
22+
&& test -x /opt/rocm/bin/hipcc \
23+
&& rm -rf /tmp/gpu-asan-debs /tmp/gpu-asan-root /tmp/packages.sha256
24+
25+
# PyTorch preloads these SDK libraries by absolute path. Point those paths at
26+
# the instrumented copies too; LD_LIBRARY_PATH alone would load two LLVMs.
27+
RUN /opt/venv/bin/python - <<'PY'
28+
from pathlib import Path
29+
import rocm_sdk
30+
31+
runtime = Path('/opt/rocm-asan-10.0/lib')
32+
for name in ('amd_comgr', 'amdhip64'):
33+
target = runtime / f'lib{name}.so'
34+
assert target.is_file(), target
35+
for path in rocm_sdk.find_libraries(name):
36+
path.unlink()
37+
path.symlink_to(target)
38+
PY
2139

2240
COPY src/eval_tools /opt/aka-eval-tools/src/eval_tools
2341

2442
ENV AKA_EVAL_TOOL_FRAMEWORK_ROOT=/opt/aka-eval-tools \
2543
PYTHONPATH=/opt/aka-eval-tools \
26-
AKA_GPU_ASAN_RUNTIME_DIR=/opt/rocm-7.2.0/lib/asan \
27-
AKA_GPU_ASAN_HIP_RUNTIME=/opt/rocm-7.2.0/lib/asan/libamdhip64.so \
28-
AKA_GPU_ASAN_HOST_PRELOAD=/opt/rocm-7.2.0/lib/llvm/lib/clang/22/lib/linux/libclang_rt.asan-x86_64.so \
29-
AKA_GPU_ASAN_HOST_LIB_DIR=/opt/rocm-7.2.0/lib/llvm/lib/clang/22/lib/linux \
30-
AKA_GPU_ASAN_NORMAL_ROCM_LIB_DIR=/opt/rocm-7.2.0/lib
44+
AKA_GPU_ASAN_RUNTIME_DIR=/opt/rocm-asan-10.0/lib \
45+
AKA_GPU_ASAN_HIP_RUNTIME=/opt/rocm-asan-10.0/lib/libamdhip64.so \
46+
AKA_GPU_ASAN_HOST_PRELOAD=/opt/rocm-asan-10.0/lib/llvm/lib/clang/23/lib/linux/libclang_rt.asan-x86_64.so \
47+
AKA_GPU_ASAN_HOST_LIB_DIR=/opt/rocm-asan-10.0/lib/llvm/lib/clang/23/lib/linux \
48+
AKA_GPU_ASAN_NORMAL_ROCM_LIB_DIR=/opt/rocm/lib

‎docker/eval-tools/gpu-asan/packages.sha256‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
3bd5b98b3ae2cb8fbfd10c248682feb86d0f5136914e7a226b701340f9b90f83 rocm-core-asan_7.2.0.70200-43~22.04_amd64.deb
2-
31118ea2dc79fe9d8c69ad7ff2176a2f1c822128ece604d4438b6c13a1aa3179 comgr-asan_3.0.0.70200-43~22.04_amd64.deb
3-
c5d6e48846f6163b5c8d7168949e1c24dc41f69bd871705621b0972c2f1a03cc hsa-rocr-asan_1.18.0.70200-43~22.04_amd64.deb
4-
fa56c2192f28adb022dc323d2836dfbe4f12211287575ea27978dc96b10a5300 hip-runtime-amd-asan_7.2.26015.70200-43~22.04_amd64.deb
1+
030fc18229683369c77e7c9e124dd1b80114e7708a5ba8e86b985b031c2cda96 amdrocm-base-asan10.0_10.0.0-4_amd64.deb
2+
3f45a0f26b4c9c01741fea8d42dec83f16dae4f754253d09fba32f1c32aec724 amdrocm-llvm-asan10.0_10.0.0-4_amd64.deb
3+
c23ff5640c087457980d0d8e9edda49b3eb1c2114dd21fe437ffcf5e942fd15b amdrocm-runtime-asan10.0_10.0.0-4_amd64.deb
4+
ff3c7cecf40d8133692116455266a7c0c02a2bdf5b695fdae746e47e50ff7147 amdrocm-sysdeps-asan10.0_10.0.0-4_amd64.deb

‎docker/eval-tools/hip-fpsan/Dockerfile‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
ARG BASE_IMAGE=lmsysorg/sglang-rocm:v0.5.14-rocm720-mi35x-20260705@sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78
1+
ARG BASE_IMAGE=lmsysorg/sglang:v0.5.20-rocm10-mi35x@sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69
22
FROM ${BASE_IMAGE}
33

44
ARG HIP_FPSAN_COMMIT=0ac9be8a1539a473ba21dfa686564c3be33c890e
55

66
LABEL org.opencontainers.image.title="AgentKernelArena HIP-FpSan tool runtime" \
7-
org.opencontainers.image.base.digest="sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78" \
7+
org.opencontainers.image.base.digest="sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69" \
88
org.opencontainers.image.revision="0ac9be8a1539a473ba21dfa686564c3be33c890e"
99

1010
COPY src/eval_tools/probes/hip_fpsan_probe.hip /tmp/hip_fpsan_probe.hip

‎docker/eval-tools/images.lock.yaml‎

Lines changed: 28 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -1,36 +1,41 @@
11
schema_version: 1
22

3-
# The scoring runtime remains unchanged. Every tool image is an immutable child
4-
# of this verified MI355X/gfx950 image.
3+
# Every tool runtime is a child of the same pinned MI355X/gfx950 scoring base.
4+
# Native rocJITsu build dependencies have a separate builder lock below.
55
base:
66
gfx950:
7-
reference: lmsysorg/sglang-rocm:v0.5.14-rocm720-mi35x-20260705
8-
digest: sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78
9-
rocm: 7.2.0
7+
reference: lmsysorg/sglang:v0.5.20-rocm10-mi35x
8+
digest: sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69
9+
rocm: 10.0.0
10+
11+
# The native C++ engines keep their GCC 13/Ubuntu 22.04 build toolchain.
12+
# Only installed engines and their C++ runtime libraries enter the ROCm 10
13+
# sidecars; HIP candidate/control compilation uses the new runtime SDK.
14+
builders:
15+
rocjitsu:
16+
reference: lmsysorg/sglang-rocm@sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78
1017

1118
tools:
1219
triton_fpsan:
1320
triton:
14-
version: 3.7.0+amd.rocm7.2.0.gitd0d77a509
15-
sha256: 3a8acacdfb4723c8bb71844c427f4d3ce047658bf97fbdc23065c06205231687
16-
triton_kernels:
17-
version: 1.0.0+amd.rocm7.2.0.gitd0d77a509
18-
sha256: df8b42ebf098767c0d31a3916913bd7557a623903aefa26602e7c6818c34b84a
21+
version: 3.8.0+git4cff872c.rocm10.0.0
22+
source: pinned_base_image
1923

2024
gpu_asan:
25+
repository: https://stable.repo.amd.com/rocm/core/packages-asan/ubuntu2404
2126
packages:
22-
rocm-core-asan:
23-
version: 7.2.0.70200-43~22.04
24-
sha256: 3bd5b98b3ae2cb8fbfd10c248682feb86d0f5136914e7a226b701340f9b90f83
25-
comgr-asan:
26-
version: 3.0.0.70200-43~22.04
27-
sha256: 31118ea2dc79fe9d8c69ad7ff2176a2f1c822128ece604d4438b6c13a1aa3179
28-
hsa-rocr-asan:
29-
version: 1.18.0.70200-43~22.04
30-
sha256: c5d6e48846f6163b5c8d7168949e1c24dc41f69bd871705621b0972c2f1a03cc
31-
hip-runtime-amd-asan:
32-
version: 7.2.26015.70200-43~22.04
33-
sha256: fa56c2192f28adb022dc323d2836dfbe4f12211287575ea27978dc96b10a5300
27+
amdrocm-base-asan10.0:
28+
version: 10.0.0-4
29+
sha256: 030fc18229683369c77e7c9e124dd1b80114e7708a5ba8e86b985b031c2cda96
30+
amdrocm-llvm-asan10.0:
31+
version: 10.0.0-4
32+
sha256: 3f45a0f26b4c9c01741fea8d42dec83f16dae4f754253d09fba32f1c32aec724
33+
amdrocm-runtime-asan10.0:
34+
version: 10.0.0-4
35+
sha256: c23ff5640c087457980d0d8e9edda49b3eb1c2114dd21fe437ffcf5e942fd15b
36+
amdrocm-sysdeps-asan10.0:
37+
version: 10.0.0-4
38+
sha256: ff3c7cecf40d8133692116455266a7c0c02a2bdf5b695fdae746e47e50ff7147
3439

3540
rocjitsu:
3641
repository: https://github.com/ROCm/rocm-systems.git
@@ -70,5 +75,5 @@ tools:
7075
image_probe: src/eval_tools/probes/hip_fpsan_probe.hip
7176

7277
verification:
73-
gfx950: verified
78+
gfx950: qualification_pending
7479
gfx942: unverified

‎docker/eval-tools/rocjitsu-sanitizers/Dockerfile‎

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
1-
ARG BASE_IMAGE=lmsysorg/sglang-rocm:v0.5.14-rocm720-mi35x-20260705@sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78
2-
FROM ${BASE_IMAGE} AS builder
1+
ARG BASE_IMAGE=lmsysorg/sglang:v0.5.20-rocm10-mi35x@sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69
2+
# Keep the native engine build toolchain pinned independently of its runtime.
3+
ARG BUILDER_IMAGE=lmsysorg/sglang-rocm:v0.5.14-rocm720-mi35x-20260705@sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78
4+
FROM ${BUILDER_IMAGE} AS builder
5+
ARG BUILD_JOBS=8
36

47
ARG ROCJITSU_SANITIZERS_COMMIT=ed35c0b54547c98bab359c8732529d9f5e8fd1ae
58
ARG GOOGLETEST_COMMIT=b514bdc898e2951020cbdca1304b75f5950d1f59
@@ -45,7 +48,7 @@ RUN curl -sSfL -o /tmp/zstd.tar.gz \
4548
-DZSTD_BUILD_STATIC=ON \
4649
-DZSTD_BUILD_PROGRAMS=OFF \
4750
-DZSTD_BUILD_TESTS=OFF \
48-
&& cmake --build /build/zstd --parallel \
51+
&& cmake --build /build/zstd --parallel "${BUILD_JOBS}" \
4952
&& cmake --install /build/zstd \
5053
&& rm -f /tmp/zstd.tar.gz
5154

@@ -58,7 +61,7 @@ RUN cmake -S /src/rocm-systems/emulation/rocjitsu -B /build/rocjitsu -G Ninja \
5861
-DFETCHCONTENT_SOURCE_DIR_GOOGLETEST=/src/googletest \
5962
-DFETCHCONTENT_SOURCE_DIR_FLATBUFFERS=/src/flatbuffers \
6063
-DBUILD_TESTING=OFF \
61-
&& cmake --build /build/rocjitsu --parallel \
64+
&& cmake --build /build/rocjitsu --parallel "${BUILD_JOBS}" \
6265
--target rj_waitcheck rocjitsu_waitcheck rocjitsu_dbi_hooks \
6366
&& mkdir -p /opt/rocjitsu/include /opt/rocjitsu/lib \
6467
&& cp -a /src/rocm-systems/emulation/rocjitsu/lib/rocjitsu/include/rocjitsu \
@@ -81,15 +84,15 @@ RUN g++-13 -std=c++17 -O2 -Wall -Wextra -Werror \
8184

8285
FROM ${BASE_IMAGE} AS runtime-common
8386

84-
LABEL org.opencontainers.image.base.digest="sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78" \
87+
LABEL org.opencontainers.image.base.digest="sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69" \
8588
org.opencontainers.image.revision="ed35c0b54547c98bab359c8732529d9f5e8fd1ae"
8689

8790
COPY --from=builder /opt/rocjitsu /opt/rocjitsu
8891
COPY src/eval_tools /opt/aka-eval-tools/src/eval_tools
8992

9093
ENV AKA_EVAL_TOOL_FRAMEWORK_ROOT=/opt/aka-eval-tools \
9194
PYTHONPATH=/opt/aka-eval-tools \
92-
LD_LIBRARY_PATH=/opt/rocjitsu/lib:/opt/rocjitsu/runtime \
95+
LD_LIBRARY_PATH=/opt/rocjitsu/lib:/opt/rocjitsu/runtime:${LD_LIBRARY_PATH} \
9396
PATH=/opt/rocjitsu/bin:${PATH} \
9497
AKA_ROCJITSU_SANITIZERS_COMMIT=ed35c0b54547c98bab359c8732529d9f5e8fd1ae
9598

@@ -112,6 +115,7 @@ LABEL org.opencontainers.image.title="AgentKernelArena rocJITsu ConSan runtime"
112115
ENV AKA_CONSAN_HOOK=/opt/rocjitsu/lib/librocjitsu_dbi_hooks.so
113116

114117
RUN test -f /opt/rocjitsu/lib/librocjitsu_dbi_hooks.so \
118+
&& /opt/venv/bin/python -c "import ctypes; ctypes.CDLL('libamdhip64.so.7')" \
115119
&& /opt/venv/bin/python -I \
116120
/opt/aka-eval-tools/src/eval_tools/adapters/consan_entrypoint.py \
117121
--help >/dev/null

‎docker/eval-tools/rocjitsu/Dockerfile‎

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
1-
ARG BASE_IMAGE=lmsysorg/sglang-rocm:v0.5.14-rocm720-mi35x-20260705@sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78
2-
FROM ${BASE_IMAGE} AS builder
1+
ARG BASE_IMAGE=lmsysorg/sglang:v0.5.20-rocm10-mi35x@sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69
2+
# Keep the native engine build toolchain pinned independently of its runtime.
3+
ARG BUILDER_IMAGE=lmsysorg/sglang-rocm:v0.5.14-rocm720-mi35x-20260705@sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78
4+
FROM ${BUILDER_IMAGE} AS builder
5+
ARG BUILD_JOBS=8
36

47
ARG ROCJITSU_COMMIT=0bf561a0d8a4a6b88954f2c46bd3a50871cda140
58
ARG GOOGLETEST_COMMIT=b514bdc898e2951020cbdca1304b75f5950d1f59
@@ -40,7 +43,7 @@ RUN cmake -S /src/rocm-systems/emulation/rocjitsu -B /build/rocjitsu -G Ninja \
4043
-DCMAKE_HIP_ARCHITECTURES=gfx950 \
4144
-DBUILD_TESTING=ON \
4245
-DRJ_INSTALL_TESTS=OFF \
43-
&& cmake --build /build/rocjitsu --parallel \
46+
&& cmake --build /build/rocjitsu --parallel "${BUILD_JOBS}" \
4447
&& cmake --install /build/rocjitsu \
4548
&& mkdir -p /opt/rocjitsu/runtime \
4649
&& cp -a /usr/lib/x86_64-linux-gnu/libstdc++.so.6* /opt/rocjitsu/runtime/ \
@@ -49,7 +52,7 @@ RUN cmake -S /src/rocm-systems/emulation/rocjitsu -B /build/rocjitsu -G Ninja \
4952
FROM ${BASE_IMAGE}
5053

5154
LABEL org.opencontainers.image.title="AgentKernelArena rocJITsu tool runtime" \
52-
org.opencontainers.image.base.digest="sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78" \
55+
org.opencontainers.image.base.digest="sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69" \
5356
org.opencontainers.image.revision="0bf561a0d8a4a6b88954f2c46bd3a50871cda140"
5457

5558
COPY --from=builder /opt/rocjitsu /opt/rocjitsu

‎docker/eval-tools/triton-fpsan/Dockerfile‎

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,16 @@
1-
ARG BASE_IMAGE=lmsysorg/sglang-rocm:v0.5.14-rocm720-mi35x-20260705@sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78
1+
ARG BASE_IMAGE=lmsysorg/sglang:v0.5.20-rocm10-mi35x@sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69
22
FROM ${BASE_IMAGE}
33

44
LABEL org.opencontainers.image.title="AgentKernelArena Triton FpSan tool runtime" \
5-
org.opencontainers.image.base.digest="sha256:b435b508b5aa696abb25c909341ce73e41574c4271cf716bed72418dcea86b78" \
6-
org.opencontainers.image.version="triton-3.7.0+amd.rocm7.2.0.gitd0d77a509"
5+
org.opencontainers.image.base.digest="sha256:e20849665c105d389ef91d23c0dc73931aaa6f02056dd10e7b43e4f16c79df69" \
6+
org.opencontainers.image.version="triton-3.8.0+git4cff872c.rocm10.0.0"
77

8-
COPY docker/eval-tools/triton-fpsan/requirements.lock /tmp/eval-tool-requirements.lock
9-
10-
RUN python -m pip uninstall -y \
11-
triton pytorch-triton pytorch-triton-rocm triton-rocm amd-triton triton-kernels \
12-
&& python -m pip install --no-cache-dir --no-deps --require-hashes \
13-
-r /tmp/eval-tool-requirements.lock \
14-
&& python -c "import importlib.metadata as m, triton, triton_kernels; assert m.version('triton') == '3.7.0+amd.rocm7.2.0.gitd0d77a509'; assert m.version('triton-kernels') == '1.0.0+amd.rocm7.2.0.gitd0d77a509'" \
15-
&& rm -f /tmp/eval-tool-requirements.lock
8+
# The digest-pinned ROCm 10 base already supplies FpSan-enabled Triton.
9+
# Do not replace it with a ROCm 7.2 wheel or infer support from import success.
10+
RUN python -c "import importlib.metadata as m; assert m.version('triton') == '3.8.0+git4cff872c.rocm10.0.0'"
1611

1712
COPY src/eval_tools /opt/aka-eval-tools/src/eval_tools
1813

1914
ENV AKA_EVAL_TOOL_FRAMEWORK_ROOT=/opt/aka-eval-tools \
2015
PYTHONPATH=/opt/aka-eval-tools \
21-
AKA_TRITON_FPSAN_VERSION=3.7.0+amd.rocm7.2.0.gitd0d77a509
16+
AKA_TRITON_FPSAN_VERSION=3.8.0+git4cff872c.rocm10.0.0

‎docker/eval-tools/triton-fpsan/requirements.lock‎

Lines changed: 0 additions & 4 deletions
This file was deleted.

0 commit comments

Comments
 (0)