diff --git a/qa/L0_backend_python/test.sh b/qa/L0_backend_python/test.sh index 587d1b8e13..dd08b68369 100755 --- a/qa/L0_backend_python/test.sh +++ b/qa/L0_backend_python/test.sh @@ -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. diff --git a/qa/L0_backend_python/unittest/test.sh b/qa/L0_dlpack_multi_gpu/test.sh similarity index 87% rename from qa/L0_backend_python/unittest/test.sh rename to qa/L0_dlpack_multi_gpu/test.sh index 411e2e76bc..af528a6667 100644 --- a/qa/L0_backend_python/unittest/test.sh +++ b/qa/L0_dlpack_multi_gpu/test.sh @@ -1,5 +1,5 @@ #!/bin/bash -# Copyright 2021-2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# Copyright 2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions @@ -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=./python_unittest.py CLIENT_LOG="./client.log" EXPECTED_NUM_TESTS="1" TEST_RESULT_FILE='test_results.txt' @@ -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 @@ -50,8 +50,10 @@ 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 +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