Skip to content

[feat] : store additional drivers enabled/disabled list on host #529

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
rahulait wants to merge 1 commit into
base: main
Choose a base branch
from
Open

[feat] : store additional drivers enabled/disabled list on host #529

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 16 additions & 0 deletions azurelinux/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 &
Expand Down
16 changes: 16 additions & 0 deletions centos7/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down Expand Up @@ -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 &
Expand Down
16 changes: 16 additions & 0 deletions centos8/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down Expand Up @@ -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 &
Expand Down
16 changes: 16 additions & 0 deletions coreos/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand All @@ -275,6 +290,7 @@ init() {
_install_driver
_load_driver
_mount_rootfs
_store_additional_drivers_flags

echo "Done, now waiting for signal"
sleep infinity &
Expand Down
16 changes: 16 additions & 0 deletions fedora/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 &
Expand Down
16 changes: 16 additions & 0 deletions flatcar/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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}"
Expand All @@ -391,6 +406,7 @@ init() {
_install_driver
_load_driver
_mount_rootfs
_store_additional_drivers_flags

echo "Done, now waiting for signal"
sleep infinity &
Expand Down
16 changes: 16 additions & 0 deletions photon3.0/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down Expand Up @@ -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 &
Expand Down
16 changes: 16 additions & 0 deletions rhel7/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down Expand Up @@ -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 &
Expand Down
16 changes: 16 additions & 0 deletions rhel8/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 &
Expand Down
16 changes: 16 additions & 0 deletions rhel9/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 &
Expand Down
16 changes: 16 additions & 0 deletions sle15/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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 &
Expand Down
16 changes: 16 additions & 0 deletions ubuntu16.04/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down Expand Up @@ -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 &
Expand Down
16 changes: 16 additions & 0 deletions ubuntu18.04/nvidia-driver
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down Expand Up @@ -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 &
Expand Down
Loading