From e3a08c10d9c1055285f1dee68702dbd7f04e2e6e Mon Sep 17 00:00:00 2001
From: Rahul Sharma <rahulsharm@nvidia.com>
Date: Wed, 24 Dec 2025 15:58:00 -0800
Subject: [PATCH] store additional drivers enabled/disabled list on host

Signed-off-by: Rahul Sharma <rahulsharm@nvidia.com>
---
 azurelinux/nvidia-driver  | 16 ++++++++++++++++
 centos7/nvidia-driver     | 16 ++++++++++++++++
 centos8/nvidia-driver     | 16 ++++++++++++++++
 coreos/nvidia-driver      | 16 ++++++++++++++++
 fedora/nvidia-driver      | 16 ++++++++++++++++
 flatcar/nvidia-driver     | 16 ++++++++++++++++
 photon3.0/nvidia-driver   | 16 ++++++++++++++++
 rhel7/nvidia-driver       | 16 ++++++++++++++++
 rhel8/nvidia-driver       | 16 ++++++++++++++++
 rhel9/nvidia-driver       | 16 ++++++++++++++++
 sle15/nvidia-driver       | 16 ++++++++++++++++
 ubuntu16.04/nvidia-driver | 16 ++++++++++++++++
 ubuntu18.04/nvidia-driver | 16 ++++++++++++++++
 ubuntu20.04/nvidia-driver | 16 ++++++++++++++++
 ubuntu22.04/nvidia-driver | 16 ++++++++++++++++
 ubuntu24.04/nvidia-driver | 16 ++++++++++++++++
 16 files changed, 256 insertions(+)

diff --git a/azurelinux/nvidia-driver b/azurelinux/nvidia-driver
index 19d2b31ab..7b7b0c8ee 100755
--- a/azurelinux/nvidia-driver
+++ b/azurelinux/nvidia-driver
@@ -265,10 +265,26 @@ _shutdown() {
     return 1
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 _load() {
     _install_driver
     _load_driver
     _mount_rootfs
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/centos7/nvidia-driver b/centos7/nvidia-driver
index a185906f6..d28981a99 100755
--- a/centos7/nvidia-driver
+++ b/centos7/nvidia-driver
@@ -410,6 +410,21 @@ _shutdown() {
     return 1
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     echo -e "\n========== NVIDIA Software Installer ==========\n"
     echo -e "Starting installation of NVIDIA driver version ${DRIVER_VERSION} for Linux kernel version ${KERNEL_VERSION}\n"
@@ -440,6 +455,7 @@ init() {
     _load_driver
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/centos8/nvidia-driver b/centos8/nvidia-driver
index 3dea528dc..3df19246f 100755
--- a/centos8/nvidia-driver
+++ b/centos8/nvidia-driver
@@ -274,6 +274,21 @@ _shutdown() {
     return 1
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     echo -e "\n========== NVIDIA Software Installer ==========\n"
     echo -e "Starting installation of NVIDIA driver version ${DRIVER_VERSION} for Linux kernel version ${KERNEL_VERSION}\n"
@@ -304,6 +319,7 @@ init() {
     _load_driver
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/coreos/nvidia-driver b/coreos/nvidia-driver
index 10d690a1f..24f50f4c9 100755
--- a/coreos/nvidia-driver
+++ b/coreos/nvidia-driver
@@ -250,6 +250,21 @@ _shutdown() {
     return 1
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     echo -e "\n========== NVIDIA Software Installer ==========\n"
     echo -e "Starting installation of NVIDIA driver version ${DRIVER_VERSION} for Linux kernel version ${KERNEL_VERSION}\n"
@@ -275,6 +290,7 @@ init() {
     _install_driver
     _load_driver
     _mount_rootfs
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/fedora/nvidia-driver b/fedora/nvidia-driver
index a8468ad8e..f511fa22e 100755
--- a/fedora/nvidia-driver
+++ b/fedora/nvidia-driver
@@ -644,10 +644,26 @@ _build() {
     _install_driver
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 _load() {
     _load_driver
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/flatcar/nvidia-driver b/flatcar/nvidia-driver
index cf29ae1be..2b5cc2835 100755
--- a/flatcar/nvidia-driver
+++ b/flatcar/nvidia-driver
@@ -366,6 +366,21 @@ _shutdown() {
     return 1
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     printf "\\n========== NVIDIA Software Installer ==========\\n"
     printf "Starting installation of NVIDIA driver version %s for Linux kernel version %s\\n" "${DRIVER_VERSION}" "${KERNEL_VERSION}"
@@ -391,6 +406,7 @@ init() {
     _install_driver
     _load_driver
     _mount_rootfs
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/photon3.0/nvidia-driver b/photon3.0/nvidia-driver
index c95236774..ea28e6603 100755
--- a/photon3.0/nvidia-driver
+++ b/photon3.0/nvidia-driver
@@ -260,6 +260,21 @@ _shutdown() {
     return 1
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     echo -e "\n========== NVIDIA Software Installer ==========\n"
     echo -e "Starting installation of NVIDIA driver version ${DRIVER_VERSION} for Linux kernel version ${KERNEL_VERSION}\n"
@@ -290,6 +305,7 @@ init() {
     _load_driver
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/rhel7/nvidia-driver b/rhel7/nvidia-driver
index 528c1632d..c1bc056f1 100755
--- a/rhel7/nvidia-driver
+++ b/rhel7/nvidia-driver
@@ -295,6 +295,21 @@ _shutdown() {
     return 1
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     echo -e "\n========== NVIDIA Software Installer ==========\n"
     echo -e "Starting installation of NVIDIA driver version ${DRIVER_VERSION} for Linux kernel version ${KERNEL_VERSION}\n"
@@ -325,6 +340,7 @@ init() {
     _load_driver
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/rhel8/nvidia-driver b/rhel8/nvidia-driver
index ce11d4ed7..1976ff196 100755
--- a/rhel8/nvidia-driver
+++ b/rhel8/nvidia-driver
@@ -737,10 +737,26 @@ _build() {
     _install_driver
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 _load() {
     _load_driver
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/rhel9/nvidia-driver b/rhel9/nvidia-driver
index 8ecd8b1e7..4affd67bf 100755
--- a/rhel9/nvidia-driver
+++ b/rhel9/nvidia-driver
@@ -754,10 +754,26 @@ _build() {
     _install_driver
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 _load() {
     _load_driver
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/sle15/nvidia-driver b/sle15/nvidia-driver
index 57c96e201..37e7e85ea 100755
--- a/sle15/nvidia-driver
+++ b/sle15/nvidia-driver
@@ -419,6 +419,21 @@ _start_vgpu_topology_daemon() {
     nvidia-topologyd
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     if [ "${DRIVER_TYPE}" = "vgpu" ]; then
         _find_vgpu_driver_version || exit 1
@@ -464,6 +479,7 @@ init() {
     _load_driver || exit 1
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/ubuntu16.04/nvidia-driver b/ubuntu16.04/nvidia-driver
index 7786b22df..7595301e0 100755
--- a/ubuntu16.04/nvidia-driver
+++ b/ubuntu16.04/nvidia-driver
@@ -277,6 +277,21 @@ _shutdown() {
     return 1
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     echo -e "\n========== NVIDIA Software Installer ==========\n"
     echo -e "Starting installation of NVIDIA driver version ${DRIVER_VERSION} for Linux kernel version ${KERNEL_VERSION}\n"
@@ -307,6 +322,7 @@ init() {
     _load_driver
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/ubuntu18.04/nvidia-driver b/ubuntu18.04/nvidia-driver
index c61c8454c..b67695f7e 100755
--- a/ubuntu18.04/nvidia-driver
+++ b/ubuntu18.04/nvidia-driver
@@ -348,6 +348,21 @@ _shutdown() {
     return 1
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     echo -e "\n========== NVIDIA Software Installer ==========\n"
     echo -e "Starting installation of NVIDIA driver version ${DRIVER_VERSION} for Linux kernel version ${KERNEL_VERSION}\n"
@@ -378,6 +393,7 @@ init() {
     _load_driver || exit 1
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/ubuntu20.04/nvidia-driver b/ubuntu20.04/nvidia-driver
index 248e7f4e8..c800baa61 100755
--- a/ubuntu20.04/nvidia-driver
+++ b/ubuntu20.04/nvidia-driver
@@ -561,6 +561,21 @@ _start_vgpu_topology_daemon() {
     nvidia-topologyd
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     if [ "${DRIVER_TYPE}" = "vgpu" ]; then
         _find_vgpu_driver_version || exit 1
@@ -621,6 +636,7 @@ init() {
     _load_driver || exit 1
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/ubuntu22.04/nvidia-driver b/ubuntu22.04/nvidia-driver
index da3ec8e31..0ed5d9d57 100755
--- a/ubuntu22.04/nvidia-driver
+++ b/ubuntu22.04/nvidia-driver
@@ -638,6 +638,21 @@ _start_vgpu_topology_daemon() {
     nvidia-topologyd
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     if [ "${DRIVER_TYPE}" = "vgpu" ]; then
         _find_vgpu_driver_version || exit 1
@@ -699,6 +714,7 @@ init() {
     _load_driver || exit 1
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &
diff --git a/ubuntu24.04/nvidia-driver b/ubuntu24.04/nvidia-driver
index 2449628fc..26b947d74 100755
--- a/ubuntu24.04/nvidia-driver
+++ b/ubuntu24.04/nvidia-driver
@@ -566,6 +566,21 @@ _start_vgpu_topology_daemon() {
     nvidia-topologyd
 }
 
+_build_features_config() {
+    cat <<EOF
+GDRCOPY_ENABLED: ${GDRCOPY_ENABLED:-false}
+GDS_ENABLED: ${GDS_ENABLED:-false}
+GPU_DIRECT_RDMA_ENABLED: ${GPU_DIRECT_RDMA_ENABLED:-false}
+EOF
+}
+
+_store_additional_drivers_flags() {
+    local config_file="${RUN_DIR}/driver/.additional-drivers-flags"
+    echo "Storing additional drivers flags..."
+    _build_features_config > "$config_file"
+    echo "Additional drivers flags stored at $config_file"
+}
+
 init() {
     if [ "${DRIVER_TYPE}" = "vgpu" ]; then
         _find_vgpu_driver_version || exit 1
@@ -600,6 +615,7 @@ init() {
     _load_driver || exit 1
     _mount_rootfs
     _write_kernel_update_hook
+    _store_additional_drivers_flags
 
     echo "Done, now waiting for signal"
     sleep infinity &