Skip to content

Commit

Permalink
Add unit test reports to L0_dlpack_multi_gpu and L0_warmup (#6873)
Browse files Browse the repository at this point in the history
* Add unit test reports to L0_dlpack_multi_gpu

* Add unit test reports to L0_warmup
  • Loading branch information
krishung5 authored Feb 14, 2024
1 parent 4a398b6 commit c048a9f
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 21 deletions.
16 changes: 4 additions & 12 deletions qa/L0_dlpack_multi_gpu/test.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#!/bin/bash
# Copyright 2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# Copyright 2023-2024, 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
Expand Down Expand Up @@ -64,18 +64,11 @@ fi

set +e
export MODEL_NAME="dlpack_test"
python3 $CLIENT_PY > $CLIENT_LOG 2>&1
python3 -m pytest --junitxml=dlpack_multi_gpu.report.xml $CLIENT_PY > $CLIENT_LOG 2>&1

if [ $? -ne 0 ]; then
echo -e "\n***\n*** python_unittest.py FAILED. \n***"
RET=1
else
check_test_results $TEST_RESULT_FILE $EXPECTED_NUM_TESTS
if [ $? -ne 0 ]; then
cat $CLIENT_LOG
echo -e "\n***\n*** Test Result Verification Failed\n***"
RET=1
fi
fi
set -e

Expand All @@ -85,10 +78,9 @@ wait $SERVER_PID
if [ $RET -eq 1 ]; then
cat $CLIENT_LOG
cat $SERVER_LOG
echo -e "\n***\n*** Unittest test FAILED. \n***"
echo -e "\n***\n*** dlpack_multi_gpu test FAILED. \n***"
else
echo -e "\n***\n*** Unittest test PASSED. \n***"
echo -e "\n***\n*** dlpack_multi_gpu test PASSED. \n***"
fi

export CUDA_VISIBLE_DEVICES=0
exit $RET
11 changes: 2 additions & 9 deletions qa/L0_warmup/test.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#!/bin/bash
# Copyright 2019-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# Copyright 2019-2024, 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
Expand Down Expand Up @@ -462,18 +462,11 @@ fi
set +e

export MODEL_NAME='bls_onnx_warmup'
python3 $CLIENT_PY >> $CLIENT_LOG 2>&1
python3 -m pytest --junitxml=warmup.report.xml $CLIENT_PY >> $CLIENT_LOG 2>&1
if [ $? -ne 0 ]; then
echo -e "\n***\n*** 'bls_onnx_warmup' test FAILED. \n***"
cat $CLIENT_LOG
RET=1
else
check_test_results $TEST_RESULT_FILE $EXPECTED_NUM_TESTS
if [ $? -ne 0 ]; then
cat $CLIENT_LOG
echo -e "\n***\n*** Test Result Verification Failed\n***"
RET=1
fi
fi

set -e
Expand Down

0 comments on commit c048a9f

Please sign in to comment.