From f30281e594b1310803b4e902de4158670b4c8c8b Mon Sep 17 00:00:00 2001
From: Helena Greebe <hgreebe@amazon.com>
Date: Tue, 16 Jan 2024 11:28:49 -0500
Subject: [PATCH] Change to kernel_module

---
 .../nvidia_driver/partial/_nvidia_driver_common.rb        | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/partial/_nvidia_driver_common.rb b/cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/partial/_nvidia_driver_common.rb
index e2cf98260a..63f8fd7729 100644
--- a/cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/partial/_nvidia_driver_common.rb
+++ b/cookbooks/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