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 model_path="$1"
|
||||||
local env="$2"
|
local env="$2"
|
||||||
local suffix="$3"
|
local suffix="$3"
|
||||||
|
local mode="$4"
|
||||||
local model_name
|
local model_name
|
||||||
model_name="$(basename "${model_path}" .gguf)"
|
model_name="$(basename "${model_path}" .gguf)"
|
||||||
local log_file="$RESULTDIR/${model_name}__${env}${suffix}__rpc.log"
|
local log_file="$RESULTDIR/${model_name}__${env}${suffix}__rpc.log"
|
||||||
@@ -215,6 +216,14 @@ run_llama_bench_rpc() {
|
|||||||
return
|
return
|
||||||
fi
|
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
|
kill_local_llamabench
|
||||||
|
|
||||||
echo
|
echo
|
||||||
@@ -283,9 +292,9 @@ run_all() {
|
|||||||
continue
|
continue
|
||||||
fi
|
fi
|
||||||
|
|
||||||
for model in "${RESOLVED_MODELS[@]}"; do
|
for model in "${RESOLVED_MODELS[@]}"; do
|
||||||
run_llama_bench_rpc "$model" "$env" "$suffix"
|
run_llama_bench_rpc "$model" "$env" "$suffix" "$mode"
|
||||||
done
|
done
|
||||||
|
|
||||||
stop_remote_rpc "$env" "$remote_pid" || true
|
stop_remote_rpc "$env" "$remote_pid" || true
|
||||||
CURRENT_REMOTE_PID=""
|
CURRENT_REMOTE_PID=""
|
||||||
|
|||||||
Reference in New Issue
Block a user