Move some QA related scripts to qa/common

pangge · Nov 27, 2018 · 7ebc711 · 7ebc711
1 parent 3af3709
commit 7ebc711
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/docs/test.rst b/docs/test.rst
@@ -43,8 +43,8 @@ Generate QA Model Repository
 The QA model repository contains some simple models that are used to
 verify the correctness of TRTIS. To generate the QA model repository::
 
-  $ cd tools
-  $ ./gen_unit_test_models
+  $ cd qa/common
+  $ ./gen_qa_model_repository
 
 This will generate the model repository in /tmp/qa_model_repository.
 The TensorRT models will be created for the GPU on the system that

diff --git a/tools/check_copyright.py → qa/common/check_copyright.py b/tools/check_copyright.py → qa/common/check_copyright.py
diff --git a/tools/gen_unit_test_models → qa/common/gen_qa_model_repository b/tools/gen_unit_test_models → qa/common/gen_qa_model_repository
@@ -57,8 +57,8 @@ mkdir -p $HOST_DESTDIR
 rm -fr $HOST_SRCDIR
 mkdir -p $HOST_SRCDIR
 
-cp ./gen_qa_model_repository.py $HOST_SRCDIR/.
-cp ../qa/common/test_util.py $HOST_SRCDIR/.
+cp ./gen_qa_models.py $HOST_SRCDIR/.
+cp ./test_util.py $HOST_SRCDIR/.
 
 C2SCRIPT=c2_gen.cmds
 TFSCRIPT=tf_gen.cmds
@@ -73,7 +73,7 @@ cat >$HOST_SRCDIR/$C2SCRIPT <<EOF
 set -e
 export CUDA_VISIBLE_DEVICES=$CUDA_DEVICE
 cd $SRCDIR
-python3 $SRCDIR/gen_qa_model_repository.py --netdef=True --models_dir=$DESTDIR
+python3 $SRCDIR/gen_qa_models.py --netdef=True --models_dir=$DESTDIR
 chown -R $(id -u):$(id -g) $DESTDIR
 EOF
 
@@ -99,7 +99,7 @@ cat >$HOST_SRCDIR/$TFSCRIPT <<EOF
 set -e
 export CUDA_VISIBLE_DEVICES=$CUDA_DEVICE
 cd $SRCDIR
-python3 $SRCDIR/gen_qa_model_repository.py --graphdef=True --savedmodel=True --models_dir=$DESTDIR
+python3 $SRCDIR/gen_qa_models.py --graphdef=True --savedmodel=True --models_dir=$DESTDIR
 chown -R $(id -u):$(id -g) $DESTDIR
 EOF
 
@@ -125,7 +125,7 @@ cat >$HOST_SRCDIR/$TRTSCRIPT <<EOF
 set -e
 export CUDA_VISIBLE_DEVICES=$CUDA_DEVICE
 cd $SRCDIR
-python3 $SRCDIR/gen_qa_model_repository.py --tensorrt=True --models_dir=$DESTDIR
+python3 $SRCDIR/gen_qa_models.py --tensorrt=True --models_dir=$DESTDIR
 chown -R $(id -u):$(id -g) $DESTDIR
 EOF
 

diff --git a/tools/gen_qa_model_repository.py → qa/common/gen_qa_models.py b/tools/gen_qa_model_repository.py → qa/common/gen_qa_models.py