Skip to content

Commit 3f8b81c

Browse files
authored
Fix datadir for x86 (#5894)
1 parent 3d8d517 commit 3f8b81c

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

qa/L0_perf_nomodel/run_test.sh

+5-1
Original file line numberDiff line numberDiff line change
@@ -47,13 +47,17 @@ TRITON_DIR=${TRITON_DIR:="/opt/tritonserver"}
4747
ARCH=${ARCH:="x86_64"}
4848
SERVER=${TRITON_DIR}/bin/tritonserver
4949
BACKEND_DIR=${TRITON_DIR}/backends
50-
DATADIR=${DATADIR:="/data/inferenceserver/${REPO_VERSION}"}
5150
MODEL_REPO="${PWD}/models"
5251
PERF_CLIENT=../clients/perf_client
5352
TF_VERSION=${TF_VERSION:=2}
5453
SERVER_ARGS="--model-repository=${MODEL_REPO} --backend-directory=${BACKEND_DIR} --backend-config=tensorflow,version=${TF_VERSION}"
5554
source ../common/util.sh
5655

56+
# DATADIR is already set in environment variable for aarch64
57+
if [ "$ARCH" != "aarch64" ]; then
58+
DATADIR="/data/inferenceserver/${REPO_VERSION}"
59+
fi
60+
5761
# Select the single GPU that will be available to the inference server
5862
export CUDA_VISIBLE_DEVICES=0
5963

0 commit comments

Comments
 (0)