mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
update
This commit is contained in:
parent
8cfe8d33a3
commit
e318c7317b
2 changed files with 3 additions and 3 deletions
2
build.sh
2
build.sh
|
|
@ -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 "$@"
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in a new issue