[Nvidia-Imex] Adding correct comments

Author: Himani Anil Deshpande

cookbooks/aws-parallelcluster-platform/attributes/platform.rb

@@ -22,7 +22,7 @@
   default['cluster']['nvidia']['driver_version'] = '550.127.08'
 end
 
-# Nvidia-imex
+# nvidia-imex
 default['cluster']['nvidia']['imex']['shared_dir'] = "#{node['cluster']['shared_dir']}/nvidia-imex"
 
 # DCV

cookbooks/aws-parallelcluster-platform/libraries/nvidia.rb

@@ -32,7 +32,5 @@ def get_device_ids
   # NVSwitch device id is 10de:22a3 for P5 instance
   # NVSwitch device id is 10de:2901 for P6 instance
   # NVSwitch device id is 10de:2941 for P6e instance
-  # We sum the count for all these deviceIds as output of lscpi command will be >0
-  # for only one device ID based on the instance type
   { 'a100' => '10de:1af1', 'h100' => '10de:22a3', 'b200' => '10de:2901', 'gb200' => '10de:2941' }
 end

cookbooks/aws-parallelcluster-platform/recipes/install/nvidia_install.rb

@@ -25,4 +25,4 @@
 
 nvidia_dcgm 'install Nvidia datacenter-gpu-manager'
 
-nvidia_imex 'Install Nvidia-imex'
+nvidia_imex 'Install nvidia-imex'

cookbooks/aws-parallelcluster-platform/resources/fabric_manager/partial/_fabric_manager_common.rb

@@ -54,6 +54,8 @@ def _nvidia_driver_version
 
 # Get number of nv switches
 def get_nvswitches
+  # We sum the count for all these deviceIds as output of lscpi command will be >0
+  # for only one device ID based on the instance type
   nvswitch_device_ids = get_device_ids.values
   nvswitch_device_ids.sum { |id| get_nvswitch_count(id) }
 end

cookbooks/aws-parallelcluster-platform/templates/nvidia-imex/nvidia-imex-nodes.erb

@@ -1,3 +1,3 @@
-## Please replace below fake IP's
+## Please replace below fake IP's with correct IP address of launched instances in Gb200 Capacity Block
 172.31.51.93
 172.31.48.43