[NVIDIA-IMEX] Configure Nvidia-imex only if we use Gb200 instance

Author: Himani Anil Deshpande

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

@@ -55,6 +55,17 @@
   end
 end
 
+action :configure do
+  return unless imex_installed
+  # Start nvidia-imex on p6e-gb200
+  if get_nvswitch_count(get_device_ids['gb200']) > 1
+    service 'nvidia-imex' do
+      action %i(start enable)
+      supports status: true
+    end unless on_docker?
+  end
+end
+
 def imex_installed
   ::File.exist?('/usr/bin/nvidia-imex') || ::File.exist?('/usr/bin/nvidia-imex-ctl')
 end

cookbooks/aws-parallelcluster-platform/test/controls/nvidia_imex_spec.rb

@@ -28,7 +28,7 @@
 
   nvidia_imex_dir = "#{node['cluster']['shared_dir']}/nvidia-imex"
 
-  %w("#{nvidia_imex_dir}/config.cfg" "#{nvidia_imex_dir}/nodes_config.cfg").each do |conf_files|
+  ["#{nvidia_imex_dir}/config.cfg", "#{nvidia_imex_dir}/nodes_config.cfg"].each do |conf_files|
     describe file(conf_files) do
       it { should exist }
       its('owner') { should eq 'root' }
@@ -39,7 +39,7 @@
 end
 
 control 'tag:config_nvidia_fabric_manager_enabled' do
-  only_if { instance.nvs_switch_enabled? }
+  only_if { instance.nvs_switch_enabled? && node['cluster']['node_type'] == "ComputeFleet" }
 
   describe service('nvidia-imex') do
     it { should be_enabled }