Skip to content

Commit 506473d

Browse files
committed
ci(gpu): repoint GPU probe image lookup at Dockerfile.gateway
The previous awk parsed `FROM <image> AS gateway` from the now-deleted `Dockerfile.images`. The new `Dockerfile.gateway` uses an ARG with a default (`ARG GATEWAY_BASE_IMAGE=nvcr.io/nvidia/distroless/cc:v4.0.4`) and `FROM ${GATEWAY_BASE_IMAGE} AS gateway`, so the old script returns nothing. Parse the ARG default value directly so the GPU prerequisites check keeps using the gateway base image as a `nvidia-smi` probe target.
1 parent bbc46d3 commit 506473d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

.github/workflows/e2e-gpu-test.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ jobs:
5858
- name: Check Docker GPU prerequisites
5959
run: |
6060
docker info --format '{{json .CDISpecDirs}}'
61-
GPU_PROBE_IMAGE="$(awk '$1 == "FROM" && $3 == "AS" && $4 == "gateway" { print $2; exit }' deploy/docker/Dockerfile.images)"
61+
GPU_PROBE_IMAGE="$(awk -F= '/^ARG[[:space:]]+GATEWAY_BASE_IMAGE=/ { print $2; exit }' deploy/docker/Dockerfile.gateway)"
6262
test -n "${GPU_PROBE_IMAGE}"
6363
docker run --rm --device nvidia.com/gpu=all "${GPU_PROBE_IMAGE}" nvidia-smi -L
6464

0 commit comments

Comments
 (0)