Skip to content

Commit

Permalink
Fix up
Browse files Browse the repository at this point in the history
  • Loading branch information
Tabrizian committed Jul 6, 2023
1 parent 647663b commit fc4d6e0
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions qa/L0_dlpack_multi_gpu/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@

SERVER=/opt/tritonserver/bin/tritonserver
SERVER_ARGS="--model-repository=`pwd`/models --log-verbose=1"
CLIENT_PY=../L0_backend_python/python_unittest.py
CLIENT_PY=./python_unittest.py
CLIENT_LOG="./client.log"
EXPECTED_NUM_TESTS="1"
TEST_RESULT_FILE='test_results.txt'
Expand All @@ -40,7 +40,7 @@ rm -fr *.log ./models
source ../common/util.sh

# Uninstall the non CUDA version of PyTorch
pip3 uninstall -y torch
# pip3 uninstall -y torch
pip3 install torch==1.13.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install tensorflow

Expand All @@ -52,6 +52,8 @@ rm -fr *.log ./models
mkdir -p models/dlpack_test/1/
cp ../python_models/dlpack_test/model.py models/dlpack_test/1/
cp ../python_models/dlpack_test/config.pbtxt models/dlpack_test
cp ../L0_backend_python/python_unittest.py .
sed -i 's#sys.path.append("../../common")#sys.path.append("../common")#g' python_unittest.py

run_server
if [ "$SERVER_PID" == "0" ]; then
Expand Down

0 comments on commit fc4d6e0

Please sign in to comment.