Browse Source

!7349 add train ut to CI

Merge pull request !7349 from lz/master
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
0cd9122e45
2 changed files with 11 additions and 0 deletions
  1. +5
    -0
      mindspore/lite/test/run_train_ut.sh
  2. +6
    -0
      mindspore/lite/test/runtest.sh

+ 5
- 0
mindspore/lite/test/run_train_ut.sh View File

@@ -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

+ 6
- 0
mindspore/lite/test/runtest.sh View File

@@ -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

Loading…
Cancel
Save