gpu-manager: enable kernel module configuration via KernelModuleConfig

Signed-off-by: Shiva Kumar (SW-CLOUD) <[email protected]>

Author: Shiva Kumar

vgpu-manager/ubuntu22.04/nvidia-driver

@@ -133,8 +133,44 @@ _set_fw_search_path() {
     echo -n "$nv_fw_search_path" > $fw_path_config_file
 }
 
-# Currently _install_driver() takes care of loading nvidia modules. Just need to start necessary vgpu daemons
+# Load NVIDIA driver kernel modules with custom parameters and start vGPU daemons
 _load_driver() {
+    local base_path="/drivers"
+    local nvidia_params=()
+    
+    # The installer may have already loaded the modules with default parameters
+    # Unload them if they're already loaded
+    if [ -f /sys/module/nvidia_vgpu_vfio/refcnt ] || [ -f /sys/module/nvidia/refcnt ]; then
+        echo "NVIDIA modules already loaded by installer, unloading to apply custom parameters..."
+        rmmod nvidia_vgpu_vfio 2>/dev/null || true
+        rmmod nvidia 2>/dev/null || true
+    fi
+    
+    echo "Parsing kernel module parameters..."
+    # Read parameters from nvidia.conf
+    if [ -f "${base_path}/nvidia.conf" ]; then
+        echo "Loading parameters from ${base_path}/nvidia.conf"
+        while IFS="" read -r line || [ -n "$line" ]; do
+            # Skip empty lines and comments
+            [[ -z "$line" || "$line" =~ ^[[:space:]]*# ]] && continue
+            # Remove "options nvidia " prefix if present
+            line=$(echo "$line" | sed 's/^[[:space:]]*options[[:space:]]\+nvidia[[:space:]]\+//')
+            if [ -n "$line" ]; then
+                nvidia_params+=("$line")
+            fi
+        done <"${base_path}/nvidia.conf"
+        if [ ${#nvidia_params[@]} -gt 0 ]; then
+            echo "Module parameters for nvidia: ${nvidia_params[@]}"
+        fi
+    fi
+
+    echo "Loading NVIDIA driver kernel modules..."
+    set -o xtrace +o nounset
+    modprobe nvidia ${nvidia_params[@]+"${nvidia_params[@]}"}
+    modprobe nvidia_vgpu_vfio
+    set +o xtrace -o nounset
+
+    # Start vGPU daemons
     /usr/bin/nvidia-vgpud
     /usr/bin/nvidia-vgpu-mgr &