This commit is contained in:
Changming Sun 2019-01-09 10:39:46 -08:00
parent 8cfe8d33a3
commit e318c7317b
2 changed files with 3 additions and 3 deletions

View file

@ -6,4 +6,4 @@
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
#requires python3.6 or higher
python3 $DIR/tools/ci_build/build.py --build_dir $DIR/build/Linux "$@"
python3 $DIR/tools/ci_build/build.py --use_openmp --build_dir $DIR/build/Linux "$@"

View file

@ -19,7 +19,7 @@ if [ $BUILD_DEVICE = "gpu" ]; then
--config Debug Release \
--skip_submodule_sync --enable_onnx_tests \
--parallel --build_shared_lib \
--use_cuda \
--use_cuda --use_openmp \
--cuda_home /usr/local/cuda \
--cudnn_home /usr/local/cudnn-$_CUDNN_VERSION/cuda --build_shared_lib $BUILD_EXTR_PAR
/home/onnxruntimedev/Release/onnx_test_runner -e cuda /data/onnx
@ -29,6 +29,6 @@ else
--use_tvm \
--skip_submodule_sync --enable_onnx_tests \
--enable_pybind \
--parallel --use_mkldnn --build_shared_lib $BUILD_EXTR_PAR
--parallel --use_openmp --use_mkldnn --build_shared_lib $BUILD_EXTR_PAR
/home/onnxruntimedev/Release/onnx_test_runner /data/onnx
fi