Add vulkan sample image

This change adds a sample image that can be used to demonstrate
vulkan functionality.

Signed-off-by: Evan Lezar <[email protected]>

Author: elezar

.common-ci.yml

@@ -81,6 +81,13 @@ stages:
     LOWER_CASE_SAMPLE: simplemultigpu
     PUSH_SAMPLE_ONLY_TAG: "true"
 
+# Define the sample targets
+.sample-vulkan:
+  variables:
+    SAMPLE: vulkan
+    LOWER_CASE_SAMPLE: vulkan
+    PUSH_SAMPLE_ONLY_TAG: "true"
+
 # Make buildx available as a docker CLI plugin
 .buildx-setup:
   before_script:
@@ -184,3 +191,11 @@ release:staging-device-query-ubuntu22.04:
   needs:
     - image-device-query-ubuntu22.04
 
+release:staging-vulkan:
+  extends:
+    - .release:staging
+    - .dist-ubuntu22.04
+    - .sample-vulkan
+  needs:
+    - image-vulkan-ubuntu22.04
+

.github/workflows/image.yaml

@@ -40,6 +40,7 @@ jobs:
         - deviceQuery
         - nvbandwidth
         - simpleMultiGPU
+        - vulkan
         exclude:
         - dist: ubi9
           sample: deviceQuery
@@ -49,6 +50,8 @@ jobs:
           sample: nvbandwidth
         - dist: ubi9
           sample: simpleMultiGPU
+        - dist: ubi9
+          sample: vulkan
 
     steps:
       - uses: actions/checkout@v4

.nvidia-ci.yml

@@ -96,6 +96,12 @@ image-simple-multi-gpu-ubuntu22.04:
     - .dist-ubuntu22.04
     - .sample-simple-multi-gpu
 
+image-vulkan-ubuntu22.04:
+  extends:
+    - .image-pull
+    - .dist-ubuntu22.04
+    - .sample-vulkan
+
 # The .scan step forms the base of the image scan operation performed before releasing
 # images.
 .scan:
@@ -223,6 +229,25 @@ scan-simple-multi-gpu-ubuntu22.04-arm64:
     - image-simple-multi-gpu-ubuntu22.04
     - scan-simple-multi-gpu-ubuntu22.04-amd64
 
+scan-vulkan-ubuntu22.04-amd64:
+  extends:
+    - .scan
+    - .sample-vulkan
+    - .dist-ubuntu22.04
+    - .platform-amd64
+  needs:
+    - image-vulkan-ubuntu22.04
+
+scan-vulkan-ubuntu22.04-arm64:
+  extends:
+    - .scan
+    - .sample-vulkan
+    - .dist-ubuntu22.04
+    - .platform-arm64
+  needs:
+    - image-vulkan-ubuntu22.04
+    - scan-vulkan-ubuntu22.04-amd64
+
 # Define external release helpers
 .release:ngc:
   extends:
@@ -264,3 +289,9 @@ release:ngc-simple-multi-gpu-ubuntu22.04:
     - .release:ngc
     - .dist-ubuntu22.04
     - .sample-simple-multi-gpu
+
+release:ngc-vulkan-ubuntu22.04:
+  extends:
+    - .release:ngc
+    - .dist-ubuntu22.04
+    - .sample-vulkan

deployments/container/Makefile

@@ -50,8 +50,20 @@ TEST_TARGETS := $(patsubst %,test-%, $(DISTRIBUTIONS))
 
 # Certain samples do not allow multi-arch images. We disable them here.
 # TODO: Does it make more sense to set this at a CI-level?
-ifeq ($(SAMPLE),nbody)
+AMD64_SAMPLES = vulkan
+ARM64_SAMPLES =
+SINGLE_ARCH_SAMPLES = nbody $(AMD64_SAMPLES) $(ARM64_SAMPLES)
+ifeq ($(SAMPLE),$(filter $(SAMPLE),$(SINGLE_ARCH_SAMPLES)))
+$(info Using single-architecture for $(SAMPLE))
 BUILD_MULTI_ARCH_IMAGES = false
+# Certain samples are only supported on AMD64.
+ifeq ($(SAMPLE),$(filter $(SAMPLE),$(AMD64_SAMPLES)))
+ARCH = amd64
+endif
+# Certain samples are only supported on AMD64.
+ifeq ($(SAMPLE),$(filter $(SAMPLE),$(ARM64_SAMPLES)))
+ARCH = arm64
+endif
 endif
 
 ifneq ($(BUILD_MULTI_ARCH_IMAGES),true)
@@ -82,7 +94,7 @@ endif
 
 build-%: DIST = $(*)
 # For the following samples, we use specific Dockerfiles:
-ifeq ($(SAMPLE),$(filter $(SAMPLE),nbody nvbandwidth))
+ifeq ($(SAMPLE),$(filter $(SAMPLE),nbody nvbandwidth vulkan))
 build-%: DOCKERFILE = $(CURDIR)/deployments/container/$(SAMPLE)/Dockerfile
 else
 build-%: DOCKERFILE = $(CURDIR)/deployments/container/Dockerfile.$(DOCKERFILE_SUFFIX)

deployments/container/vulkan/Dockerfile

@@ -0,0 +1,99 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+FROM nvcr.io/nvidia/cuda:12.8.1-base-ubuntu22.04 AS builder
+
+ENV DEBIAN_FRONTEND=noninteractive
+RUN apt-get update -y && apt-get install -y --no-install-recommends \
+    wget \
+    && \
+    rm -rf /var/lib/apt/lists/*
+
+# See instructions from https://vulkan.lunarg.com/doc/sdk/1.4.309.0/linux/getting_started_ubuntu.html
+RUN wget -qO- https://packages.lunarg.com/lunarg-signing-key-pub.asc | tee /etc/apt/trusted.gpg.d/lunarg.asc \
+    && \
+    wget -qO /etc/apt/sources.list.d/lunarg-vulkan-jammy.list http://packages.lunarg.com/vulkan/lunarg-vulkan-jammy.list \
+    && \
+    apt update -y \
+    && \
+    apt install -y --no-install-recommends vulkan-sdk \
+    && \
+    rm -rf /var/lib/apt/lists/*
+
+# TODO: Should we include this in the list above.
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    make \
+    cmake \
+    pkg-config \
+    gcc \
+    g++ \
+    && rm -rf /var/lib/apt/lists/*
+
+WORKDIR /build
+
+ARG BUILD_EXAMPLES="computeheadless renderheadless"
+ENV BUILD_EXAMPLES=$BUILD_EXAMPLES
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    libglm-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+# TODO: We should update to the official samples
+RUN git clone --depth=1 https://github.com/SaschaWillems/Vulkan.git \
+    && cd Vulkan \
+    sed -e 's|https://|git@|g' .gitmodules \
+    && \
+    git submodule sync \
+    && \
+    git submodule update \
+    && mkdir -p build && cd build \
+    && cmake -D RESOURCE_INSTALL_DIR=/cuda-samples ..  \
+    && make ${BUILD_EXAMPLES}
+
+FROM nvcr.io/nvidia/cuda:12.8.1-base-ubuntu22.04
+
+LABEL io.k8s.display-name="NVIDIA CUDA Vulkan samples"
+LABEL name="NVIDIA CUDA Vulkan samples"
+LABEL vendor="NVIDIA"
+LABEL version="N/A"
+LABEL release="N/A"
+LABEL summary="NVIDIA container to validate GPU support for Vulkan"
+LABEL description="See summary"
+
+COPY ./LICENSE ./licenses/LICENSE
+
+RUN mkdir -p /cuda-samples
+
+COPY --from=builder /build/Vulkan/build/bin/computeheadless /cuda-samples/bin/computeheadless
+COPY --from=builder /build/Vulkan/build/bin/renderheadless /cuda-samples/bin/renderheadless
+COPY --from=builder /build/Vulkan/shaders/glsl/computeheadless/ /cuda-samples/shaders/glsl/computeheadless/
+COPY --from=builder /build/Vulkan/shaders/glsl/renderheadless/ /cuda-samples/shaders/glsl/renderheadless/
+COPY --from=builder /build/Vulkan/shaders/hlsl/computeheadless/ /cuda-samples/shaders/hlsl/computeheadless/
+COPY --from=builder /build/Vulkan/shaders/hlsl/renderheadless/ /cuda-samples/shaders/hlsl/renderheadless/
+
+COPY /deployments/container/vulkan/entrypoint.sh /cuda-samples/entrypoint.sh
+RUN ln -s /cuda-samples/entrypoint.sh /cuda-samples/sample
+
+ENV DEBIAN_FRONTEND=noninteractive
+COPY --from=builder /etc/apt/sources.list.d/lunarg-vulkan-jammy.list /etc/apt/sources.list.d/lunarg-vulkan-jammy.list
+COPY --from=builder /etc/apt/trusted.gpg.d/lunarg.asc /etc/apt/trusted.gpg.d/lunarg.asc
+
+RUN apt update -y && apt install -y --no-install-recommends \
+        vulkan-sdk \
+    && \
+    rm -rf /var/lib/apt/lists/*
+
+CMD ["/cuda-samples/sample"]

deployments/container/vulkan/entrypoint.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+EXAMPLES=/cuda-samples/bin/
+for i in $(ls ${EXAMPLES}); do
+        echo 'y' | ${EXAMPLES}/$i
+        echo "\n"
+done