Disable CUDA install test on alinux with P4d instances

The nvidia-fabric-manager daemon is not installed on Amazon Linux 1 AMIs and this makes the deviceQuery utility fail on P4d instances.

Skipping the test as the behavior is expected.
Signed-off-by: ddeidda <[email protected]>

Author: ddeidda

recipes/tests.rb

@@ -378,38 +378,39 @@ module load intelmpi && mpirun --help | grep '#{node['cfncluster']['intelmpi']['
   TESTNVIDIA
 end
 
-bash 'test CUDA install' do
-  cwd Chef::Config[:file_cache_path]
-  code <<-TESTCUDA
-    has_gpu=$(lspci | grep -o "NVIDIA")
-    if [ -z "$has_gpu" ]; then
-      echo "No GPU detected, no test needed."
-      exit 0
-    fi
-
-    set -e
-    cuda_ver="#{node['cfncluster']['nvidia']['cuda_version']}"
-    # Test CUDA installation
-    echo "Testing CUDA install with nvcc..."
-    export PATH=/usr/local/cuda-$cuda_ver/bin:$PATH
-    export LD_LIBRARY_PATH=/usr/local/cuda-$cuda_ver/lib64:$LD_LIBRARY_PATH
-    # grep CUDA version from nvcc output. If CUDA is not installed nvcc command will fail
-    cuda_output=$(nvcc -V | grep -E -o "release [0-9]+.[0-9]+")
-    if [ "$cuda_output" != "release $cuda_ver" ]; then
-      echo "CUDA installed incorrectly! Installed $cuda_output but expected $cuda_ver"
-      exit 1
-    else
-      echo "CUDA nvcc test passed, $cuda_output"
-    fi
-
-    # Test deviceQuery
-    echo "Testing CUDA install with deviceQuery..."
-    /usr/local/cuda-$cuda_ver/extras/demo_suite/deviceQuery | grep -o "Result = PASS"
-    echo "CUDA deviceQuery test passed"
-    echo "Correctly installed CUDA $cuda_output"
-  TESTCUDA
+unless node['cfncluster']['cfn_base_os'] == 'alinux' && get_nvswitches > 1
+  bash 'test CUDA install' do
+    cwd Chef::Config[:file_cache_path]
+    code <<-TESTCUDA
+      has_gpu=$(lspci | grep -o "NVIDIA")
+      if [ -z "$has_gpu" ]; then
+        echo "No GPU detected, no test needed."
+        exit 0
+      fi
+  
+      set -e
+      cuda_ver="#{node['cfncluster']['nvidia']['cuda_version']}"
+      # Test CUDA installation
+      echo "Testing CUDA install with nvcc..."
+      export PATH=/usr/local/cuda-$cuda_ver/bin:$PATH
+      export LD_LIBRARY_PATH=/usr/local/cuda-$cuda_ver/lib64:$LD_LIBRARY_PATH
+      # grep CUDA version from nvcc output. If CUDA is not installed nvcc command will fail
+      cuda_output=$(nvcc -V | grep -E -o "release [0-9]+.[0-9]+")
+      if [ "$cuda_output" != "release $cuda_ver" ]; then
+        echo "CUDA installed incorrectly! Installed $cuda_output but expected $cuda_ver"
+        exit 1
+      else
+        echo "CUDA nvcc test passed, $cuda_output"
+      fi
+  
+      # Test deviceQuery
+      echo "Testing CUDA install with deviceQuery..."
+      /usr/local/cuda-$cuda_ver/extras/demo_suite/deviceQuery | grep -o "Result = PASS"
+      echo "CUDA deviceQuery test passed"
+      echo "Correctly installed CUDA $cuda_output"
+    TESTCUDA
+  end
 end
-
 ###################
 # FabricManager
 ###################