task_python_nnvm.sh 891 Bytes
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

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

echo "Running compiler test..."
python -m nose -v nnvm/tests/python/compiler || exit -1
python3 -m nose -v nnvm/tests/python/compiler || exit -1

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

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

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

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