Change to kernel_module

hgreebe · Jan 16, 2024 · f30281e · f30281e
1 parent a29b7a0
commit f30281e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...oks/aws-parallelcluster-platform/resources/nvidia_driver/partial/_nvidia_driver_common.rb b/...oks/aws-parallelcluster-platform/resources/nvidia_driver/partial/_nvidia_driver_common.rb
@@ -68,7 +68,7 @@
     cwd '/tmp'
     code <<-NVIDIA
       set -e
-      #{compiler_version} ./nvidia.run --silent --dkms --disable-nouveau --no-cc-version-check #{kernel_open}
+      #{compiler_version} ./nvidia.run --silent --dkms --disable-nouveau --no-cc-version-check #{kernel_module}
       rm -f /tmp/nvidia.run
     NVIDIA
     creates '/usr/bin/nvidia-smi'
@@ -108,9 +108,9 @@ def compiler_version
   ""
 end
 
-def kernel_open
-  if node['cluster']['nvidia']['kernel-open'] == "false"
-    ""
+def kernel_module
+  if node['cluster']['nvidia']['kernel_module'] == "kernel"
+    "-m=kernel"
   else
     "-m=kernel-open"
   end