README.md 1.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
# Inference Model UT

There are several model tests currently:
- test_ernie_text_cls.cc
- test_LeViT.cc
- test_ppyolo_mbv3.cc
- test_ppyolov2_r50vd.cc
- test_resnet50.cc
- test_resnet50_quant.cc
- test_yolov3.cc

12
To build and execute tests on Linux, simply run
13 14 15 16 17 18 19 20 21 22 23 24 25 26
```
./run.sh $PADDLE_ROOT $TURN_ON_MKL $TEST_GPU_CPU $DATA_DIR
```
To build on windows, run command with busybox
```
busybox bash ./run.sh $PADDLE_ROOT $TURN_ON_MKL $TEST_GPU_CPU $DATA_DIR
```

- After run command, it will build and execute tests and download to ${DATA_DIR} automatically.
- `$PADDLE_ROOT`: paddle library path
- `$TURN_ON_MKL`: use MKL or Openblas
- `$TEST_GPU_CPU`: test both GPU/CPU mode or only CPU mode
- `$DATA_DIR`: download data path

27
now only support 4 kinds of tests which controled by `--gtest_filter` argument, test suite name should be same as following.
28 29 30 31 32 33 34 35 36 37
- `TEST(gpu_tester_*, test_name)`
- `TEST(cpu_tester_*, test_name)`
- `TEST(mkldnn_tester_*, test_name)`
- `TEST(tensorrt_tester_*, test_name)`

skpied test suite name.
- `TEST(DISABLED_gpu_tester_*, test_name)`
- `TEST(DISABLED_cpu_tester_*, test_name)`
- `TEST(DISABLED_mkldnn_tester_*, test_name)`
- `TEST(DISABLED_tensorrt_tester_*, test_name)`