diff --git a/mindspore/lite/test/run_train_ut.sh b/mindspore/lite/test/run_train_ut.sh new file mode 100755 index 0000000000..777488acf6 --- /dev/null +++ b/mindspore/lite/test/run_train_ut.sh @@ -0,0 +1,5 @@ +#!/bin/bash +cd ./ut/src/runtime/kernel/arm || exit 1 +../../../../../../build/test/lite-test --gtest_filter=NetworkTest.efficient_net +../../../../../../build/test/lite-test --gtest_filter=NetworkTest.tuning_layer +../../../../../../build/test/lite-test --gtest_filter=NetworkTest.lenetnet diff --git a/mindspore/lite/test/runtest.sh b/mindspore/lite/test/runtest.sh new file mode 100755 index 0000000000..0d87cb2c85 --- /dev/null +++ b/mindspore/lite/test/runtest.sh @@ -0,0 +1,6 @@ +#!/bin/bash +cd ./mindspore/lite/test/ || exit 1 +if [ $1 == train ]; then + echo 'run train ut tests' + ./run_train_ut.sh +fi