Merge pull request #695 from kwozyman/rpm-args

Add support for changing Vendor and Buildhost for the driver RPM
containers · Jul 25, 2024 · f6ed0b3 · f6ed0b3
2 parents 7df349f + 57a9745
commit f6ed0b3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/training/nvidia-bootc/Containerfile b/training/nvidia-bootc/Containerfile
@@ -14,6 +14,9 @@ ARG TARGET_ARCH=''
 
 ARG DRIVER_VERSION='550.90.07'
 
+ARG VENDOR=''
+ARG RPM_HOST=''
+
 USER builder
 
 WORKDIR /home/builder
@@ -63,6 +66,8 @@ RUN if [ "${KERNEL_VERSION}" == "" ]; then \
         --define "kernel_dist ${KDIST}" \
         --define "driver ${DRIVER_VERSION}" \
         --define "driver_branch ${DRIVER_STREAM}" \
+        --define "vendor ${VENDOR:-undefined}" \
+        --define "_buildhost ${RPM_HOST:-${HOSTNAME}}" \
         -v -bb SPECS/kmod-nvidia.spec
 
 FROM ${BASEIMAGE}