[NVIDIA-IMEX] USe specific Version naming for nvidia-imex installation

Author: Himani Anil Deshpande

cookbooks/aws-parallelcluster-platform/resources/nvidia_imex/nvidia_imex_alinux2023.rb

@@ -17,3 +17,8 @@
 end
 
 use 'partial/_nvidia_imex_common.rb'
+
+def _nvidia_imex_version
+  nvidia_major_version = node['cluster']['nvidia']['driver_version'].split('.')[0]
+  "#{nvidia_major_version}-#{node['cluster']['nvidia']['driver_version']}-1"
+end

cookbooks/aws-parallelcluster-platform/resources/nvidia_imex/nvidia_imex_redhat8.rb

@@ -17,3 +17,8 @@
 end
 
 use 'partial/_nvidia_imex_common.rb'
+
+def _nvidia_imex_version
+  nvidia_major_version = node['cluster']['nvidia']['driver_version'].split('.')[0]
+  "#{nvidia_major_version}-#{node['cluster']['nvidia']['driver_version']}-1"
+end

cookbooks/aws-parallelcluster-platform/resources/nvidia_imex/nvidia_imex_rocky8.rb

@@ -17,3 +17,8 @@
 end
 
 use 'partial/_nvidia_imex_common.rb'
+
+def _nvidia_imex_version
+  nvidia_major_version = node['cluster']['nvidia']['driver_version'].split('.')[0]
+  "#{nvidia_major_version}-#{node['cluster']['nvidia']['driver_version']}-1"
+end

cookbooks/aws-parallelcluster-platform/resources/nvidia_imex/nvidia_imex_ubuntu22+.rb

@@ -17,3 +17,8 @@
 end
 
 use 'partial/_nvidia_imex_common.rb'
+
+def _nvidia_imex_version
+  nvidia_major_version = node['cluster']['nvidia']['driver_version'].split('.')[0]
+  "#{nvidia_major_version}_#{node['cluster']['nvidia']['driver_version']}-1"
+end

cookbooks/aws-parallelcluster-platform/resources/nvidia_imex/partial/_nvidia_imex_common.rb

@@ -51,7 +51,7 @@
   package 'nvidia-imex' do
     retries 3
     retry_delay 5
-    version node['cluster']['nvidia']['driver_version']
+    version _nvidia_imex_version
   end
 end
 

cookbooks/aws-parallelcluster-platform/spec/unit/resources/nvidia_imex_spec.rb

@@ -1,7 +1,7 @@
 require 'spec_helper'
 
 shared_dir = "SHARED_DIR"
-nvidia_version = "NVIDIA_VERSION"
+nvidia_version = "1.2.3"
 nvidia_imex_shared_dir = "#{shared_dir}/nvidia-imex"
 imex_binary = '/usr/bin/nvidia-imex'
 imex_ctl_binary = '/usr/bin/nvidia-imex-ctl'
@@ -169,6 +169,13 @@ def self.install(chef_run)
           end
           runner(platform: platform, version: version, step_into: ['nvidia_imex'])
         end
+        cached(:nvidia_imex_version) do
+          if %(redhat rocky).include?(platform) || platform == 'amazon' && version == '2023'
+            "1-1.2.3-1"
+          else
+            "1_1.2.3-1"
+          end
+        end
 
         before do
           chef_run.node.override['cluster']['shared_dir'] = shared_dir
@@ -199,7 +206,7 @@ def self.install(chef_run)
             is_expected.not_to install_package('nvidia-imex')
               .with(retries: 3)
               .with(retry_delay: 5)
-              .with(version: nvidia_version)
+              .with(version: nvidia_imex_version)
           end
         else
           it 'installs nvidia-imex' do
@@ -224,7 +231,7 @@ def self.install(chef_run)
             is_expected.to install_package('nvidia-imex')
               .with(retries: 3)
               .with(retry_delay: 5)
-              .with(version: nvidia_version)
+              .with(version: nvidia_imex_version)
           end
         end
       end