[TEMPORARY][B200] Add nvlsm precondition check

Author: gmarciani

cookbooks/aws-parallelcluster-platform/resources/nvidia_nvlsm/partial/_nvidia_nvlsm_common.rb

@@ -18,6 +18,7 @@
 action :install do
   return unless nvlsm_installation_enabled?
 
+  action_install_precondition
   action_install_nvlsm_dependencies
   action_install_nvlsm
 
@@ -27,6 +28,19 @@
   # node_attributes 'dump node attributes'
 end
 
+action :install_precondition do
+  bash "NVLSM Preconditions" do
+    user 'root'
+    cwd node['cluster']['sources_dir']
+    code <<-CODE
+    set -e
+    #{nvidia_nvlsm_install_preconditions_commands}
+    CODE
+    retries 3
+    retry_delay 5
+  end
+end
+
 action :install_nvlsm_dependencies do
   package nvidia_nvlsm_dependencies do
     retries 3
@@ -55,10 +69,10 @@
   bash "Install nvlsm" do
     user 'root'
     cwd node['cluster']['sources_dir']
-    code <<-NVIDIA_IMEX
+    code <<-CODE
     set -e
     #{nvidia_nvlsm_install_commands}
-    NVIDIA_IMEX
+    CODE
     retries 3
     retry_delay 5
   end
@@ -91,6 +105,10 @@ def nvidia_nvlsm_install_commands
   # OS dependent
 end
 
+def nvidia_nvlsm_install_preconditions_commands
+  # OS dependent
+end
+
 def nvidia_nvlsm_dependencies
   # OS dependent
 end

cookbooks/aws-parallelcluster-platform/resources/nvidia_nvlsm/partial/_nvidia_nvlsm_debian.rb

@@ -29,6 +29,9 @@ def nvidia_nvlsm_install_commands
 end
 
 def nvidia_nvlsm_dependencies
-  # %(linux-modules-extra-aws infiniband-diags ibutils)
-  %(infiniband-diags ibutils)
+  %(linux-modules-extra-aws infiniband-diags ibutils)
+end
+
+def nvidia_nvlsm_install_preconditions_commands
+  "apt-cache policy"
 end

cookbooks/aws-parallelcluster-platform/resources/nvidia_nvlsm/partial/_nvidia_nvlsm_rhel.rb

@@ -29,6 +29,9 @@ def nvidia_nvlsm_install_commands
 end
 
 def nvidia_nvlsm_dependencies
-  # %(kernel-modules-extra-aws infiniband-diags libibumad)
-  %(infiniband-diags libibumad)
+  %(kernel-modules-extra-aws infiniband-diags libibumad)
+end
+
+def nvidia_nvlsm_install_preconditions_commands
+  "yum repolist all"
 end