Ensure ROCBLAS_USE_HIPBLASLT is properly set remotely as well

This commit is contained in:
Donato Capitella
2025-11-17 09:33:14 +00:00
parent 17b1ec2825
commit de49d65b3c
+10 -1
View File
@@ -195,6 +195,7 @@ run_llama_bench_rpc() {
local model_path="$1"
local env="$2"
local suffix="$3"
local mode="$4"
local model_name
model_name="$(basename "${model_path}" .gguf)"
local log_file="$RESULTDIR/${model_name}__${env}${suffix}__rpc.log"
@@ -215,6 +216,14 @@ run_llama_bench_rpc() {
return
fi
if [[ "$env" == rocm* ]]; then
if [[ "$mode" == off ]]; then
client_cmd="${client_cmd/-- /-- env ROCBLAS_USE_HIPBLASLT=0 }"
else
client_cmd="${client_cmd/-- /-- env ROCBLAS_USE_HIPBLASLT=1 }"
fi
fi
kill_local_llamabench
echo
@@ -284,7 +293,7 @@ run_all() {
fi
for model in "${RESOLVED_MODELS[@]}"; do
run_llama_bench_rpc "$model" "$env" "$suffix"
run_llama_bench_rpc "$model" "$env" "$suffix" "$mode"
done
stop_remote_rpc "$env" "$remote_pid" || true