Skip to content

Commit

Permalink
Add infer run to sig_tag0 model
Browse files Browse the repository at this point in the history
  • Loading branch information
kthui committed Dec 8, 2022
1 parent 73bf2bf commit 2845bec
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions qa/L0_tf_gpu_io/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -130,6 +130,7 @@ rm -rf models && mkdir -p models
cp -r $DATADIR/qa_tf_tag_sigdef_repository/sig_tag0 models
(cd models/sig_tag0 && \
echo "optimization { execution_accelerators { gpu_execution_accelerator : [ { name : \"gpu_io\"} ] } }" >> config.pbtxt)

SERVER_ARGS="--model-repository=`pwd`/models --log-verbose=1"
SERVER_LOG="sig_tag0.serverlog"
run_server
Expand All @@ -138,6 +139,19 @@ if [ "$SERVER_PID" == "0" ]; then
cat $SERVER_LOG
exit 1
fi

CLIENT_LOG="sig_tag0.gpu.log"
$CLIENT -m sig_tag0 >> $CLIENT_LOG 2>&1
if (( $? != 0 )); then
cat $CLIENT_LOG
RET=1
fi
grep "is GPU tensor: true" $SERVER_LOG
if [ $? -ne 0 ]; then
echo -e "\n***\n*** Failed. Expected input and output are GPU tensors\n***"
RET=1
fi

kill $SERVER_PID
wait $SERVER_PID

Expand Down

0 comments on commit 2845bec

Please sign in to comment.