From 48e25ed30f308c925a13e571377a02a45d86987e Mon Sep 17 00:00:00 2001 From: Christian Sonnabend Date: Mon, 20 Jan 2025 10:49:43 +0100 Subject: [PATCH] Adding hipblaslt check --- onnxruntime.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/onnxruntime.sh b/onnxruntime.sh index 003d818317..f16c540071 100644 --- 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"