# Igneum GPU bench image, AMD lane: ROCm 6 with the OpenCL runtime (rocm-opencl) and headers, so proto-opencl/host.c
# builds with plain cc -lOpenCL and clinfo lists the gfx device. Needs /dev/kfd and /dev/dri passed in (a provider
# that offers AMD cards does that). RunPod lists no AMD consumer cards and Vast.ai none in general (approximate,
# 3 Oct 2026); this recipe is for whichever host has an RX 7900 XTX (gfx1100), or a datacenter MI300X as a stand-in.
# The tag is approximate: use the newest rocm/dev-ubuntu-22.04:6.x-complete the host's kernel driver supports.
#   docker build -f Dockerfile.rocm -t <you>/igneum-gpu-bench:rocm6 .
#   docker run --device=/dev/kfd --device=/dev/dri --group-add video -it <you>/igneum-gpu-bench:rocm6
FROM rocm/dev-ubuntu-22.04:6.3-complete
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y --no-install-recommends \
      python3 curl ca-certificates openssh-server clinfo ocl-icd-opencl-dev opencl-headers perl build-essential \
    && rm -rf /var/lib/apt/lists/* \
    && mkdir -p /run/sshd /workspace
WORKDIR /workspace
CMD ["/usr/sbin/sshd", "-D"]