Adding hipblaslt check

alisw · Jan 20, 2025 · 48e25ed · 48e25ed
1 parent 0cfeaa0
commit 48e25ed
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/onnxruntime.sh b/onnxruntime.sh
@@ -34,6 +34,7 @@ if [[ "$ALIBUILD_O2_FORCE_GPU" -eq 1 ]] || [[ "$ALIBUILD_ENABLE_HIP" -eq 1 ]] ||
   [[ -d /opt/rocm/include/rocrand ]] && \
   [[ -d /opt/rocm/include/miopen ]] && \
   [[ -d /opt/rocm/include/rccl ]] && \
+  [[ -d /opt/rocm/lib/hipblaslt ]] && \
   [[ -z "$ORT_ROCM_BUILD" ]] ) && \
   ([[ -z "$ALMA_LINUX_MAJOR_VERSION" ]] || [[ "$ALMA_LINUX_MAJOR_VERSION" -eq 9 ]]); then
   export ORT_ROCM_BUILD="1"