Ensure ROCBLAS_USE_HIPBLASLT is properly set remotely as well
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user