task_python_frontend.sh 1.78 KB
Newer Older
tqchen committed
1 2 3
#!/bin/bash

export PYTHONPATH=nnvm/python:python:topi/python
4 5
# to avoid openblas threading error
export OMP_NUM_THREADS=1
tqchen committed
6

7 8 9 10
# Rebuild cython
make cython || exit -1
make cython3 || exit -1

11
echo "Running nnvm unittest..."
tqchen committed
12 13 14
python -m nose -v nnvm/tests/python/unittest || exit -1
python3 -m nose -v nnvm/tests/python/unittest || exit -1

15
echo "Running nnvm compiler test..."
tqchen committed
16 17
python3 -m nose -v nnvm/tests/python/compiler || exit -1

18
echo "Running nnvm ONNX frontend test..."
19
python3 -m nose -v nnvm/tests/python/frontend/onnx || exit -1
tqchen committed
20

21
echo "Running nnvm MXNet frontend test..."
22
python3 -m nose -v nnvm/tests/python/frontend/mxnet || exit -1
tqchen committed
23

24
echo "Running nnvm Keras frontend test..."
25
python3 -m nose -v nnvm/tests/python/frontend/keras || exit -1
26

27
echo "Running nnvm Tensorflow frontend test..."
28
python3 -m nose -v nnvm/tests/python/frontend/tensorflow || exit -1
29

30 31 32
echo "Running nnvm CoreML frontend test..."
python3 -m nose -v nnvm/tests/python/frontend/coreml || exit -1

33 34 35 36 37
echo "Running relay MXNet frontend test..."
python3 -m nose -v tests/python/frontend/mxnet || exit -1

echo "Running relay Keras frontend test..."
python3 -m nose -v tests/python/frontend/keras || exit -1
38 39 40

echo "Running relay ONNX frondend test..."
python3 -m nose -v tests/python/frontend/onnx || exit -1
41

42 43 44
echo "Running relay CoreML frondend test..."
python3 -m nose -v tests/python/frontend/coreml || exit -1

45 46 47
echo "Running relay Tensorflow frontend test..."
python3 -m nose -v tests/python/frontend/tensorflow || exit -1

48 49
echo "Running nnvm to relay frontend test..."
python3 -m nose -v tests/python/frontend/nnvm_to_relay || exit -1
50 51 52

echo "Running relay TFLite frontend test..."
python3 -m nose -v tests/python/frontend/tflite || exit -1
53 54 55

echo "Running relay caffe2 frondend test..."
python3 -m nose -v tests/python/frontend/caffe2 || exit -1