Fixing cuda build test

Author: ChSonnabend

Common/ML/test/onnxruntime-inference/run-onnxruntime-all-eps.sh

@@ -12,13 +12,45 @@ if [[ -n ${ONNXRUNTIME_ROOT:-} && -f $ONNXRUNTIME_ROOT/etc/ort-init.sh ]]; then
   source "$ONNXRUNTIME_ROOT/etc/ort-init.sh"
 fi
 
+has_cuda_device() {
+  if command -v nvidia-smi >/dev/null 2>&1 && nvidia-smi -L >/dev/null 2>&1; then
+    return 0
+  fi
+  compgen -G "/proc/driver/nvidia/gpus/*" >/dev/null
+}
+
+has_rocm_device() {
+  if command -v rocm-smi >/dev/null 2>&1 && rocm-smi -i >/dev/null 2>&1; then
+    return 0
+  fi
+  [[ -e /dev/kfd ]] && compgen -G "/dev/dri/renderD*" >/dev/null
+}
+
 if [[ -n ${ONNXRUNTIME_INFERENCE_TEST_PROVIDERS:-} ]]; then
   IFS=', ' read -r -a PROVIDERS <<< "$ONNXRUNTIME_INFERENCE_TEST_PROVIDERS"
 else
   PROVIDERS=(cpu)
-  [[ ${ORT_MIGRAPHX_BUILD:-0} == 1 ]] && PROVIDERS+=(migraphx)
-  [[ ${ORT_CUDA_BUILD:-0} == 1 ]] && PROVIDERS+=(cuda)
-  [[ ${ORT_TENSORRT_BUILD:-0} == 1 ]] && PROVIDERS+=(tensorrt)
+  if [[ ${ORT_MIGRAPHX_BUILD:-0} == 1 ]]; then
+    if has_rocm_device; then
+      PROVIDERS+=(migraphx)
+    else
+      echo "onnxruntime-inference-test: skipping migraphx, no ROCm device detected"
+    fi
+  fi
+  if [[ ${ORT_CUDA_BUILD:-0} == 1 ]]; then
+    if has_cuda_device; then
+      PROVIDERS+=(cuda)
+    else
+      echo "onnxruntime-inference-test: skipping cuda, no CUDA device detected"
+    fi
+  fi
+  if [[ ${ORT_TENSORRT_BUILD:-0} == 1 ]]; then
+    if has_cuda_device; then
+      PROVIDERS+=(tensorrt)
+    else
+      echo "onnxruntime-inference-test: skipping tensorrt, no CUDA device detected"
+    fi
+  fi
 fi
 
 echo "onnxruntime-inference-test: selected providers: ${PROVIDERS[*]}"