Skip to content

Commit

Permalink
Move multi-GPU dlpack test to a separate L0 test
Browse files Browse the repository at this point in the history
  • Loading branch information
Tabrizian committed Jun 28, 2023
1 parent 65f7cd1 commit 60a8056
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions qa/L0_backend_python/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -369,11 +369,11 @@ and shared memory pages after starting triton equals to $current_num_pages \n***
fi

# Disable env test for Jetson since cloud storage repos are not supported
# Disable ensemble, unittest, io and bls tests for Jetson since GPU Tensors are not supported
# Disable ensemble, io and bls tests for Jetson since GPU Tensors are not supported
# Disable variants test for Jetson since already built without GPU Tensor support
# Disable decoupled test because it uses GPU tensors
if [ "$TEST_JETSON" == "0" ]; then
SUBTESTS="ensemble unittest io bls decoupled variants"
SUBTESTS="ensemble io bls decoupled variants"
for TEST in ${SUBTESTS}; do
# Run each subtest in a separate virtual environment to avoid conflicts
# between dependencies.
Expand Down
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=../python_unittest.py
CLIENT_PY=../L0_backend_python/python_unittest.py
CLIENT_LOG="./client.log"
EXPECTED_NUM_TESTS="1"
TEST_RESULT_FILE='test_results.txt'
Expand All @@ -37,7 +37,7 @@ export CUDA_VISIBLE_DEVICES=0,1,2,3
RET=0
rm -fr *.log ./models

source ../../common/util.sh
source ../common/util.sh

# Uninstall the non CUDA version of PyTorch
pip3 uninstall -y torch
Expand All @@ -50,8 +50,8 @@ pip3 install cupy-cuda12x
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 ../python_models/dlpack_test/model.py models/dlpack_test/1/
cp ../python_models/dlpack_test/config.pbtxt models/dlpack_test

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

0 comments on commit 60a8056

Please sign in to comment.