mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
34 lines
1.1 KiB
Bash
Executable file
34 lines
1.1 KiB
Bash
Executable file
#!/bin/bash
|
|
set -e -o -x
|
|
|
|
id
|
|
|
|
SCRIPT_DIR="$( dirname "${BASH_SOURCE[0]}" )"
|
|
|
|
while getopts c:d:x: parameter_Option
|
|
do case "${parameter_Option}"
|
|
in
|
|
d) BUILD_DEVICE=${OPTARG};;
|
|
x) BUILD_EXTR_PAR=${OPTARG};;
|
|
esac
|
|
done
|
|
|
|
if [ $BUILD_DEVICE = "gpu" ]; then
|
|
_CUDNN_VERSION=$(echo $CUDNN_VERSION | cut -d. -f1-2)
|
|
python3 $SCRIPT_DIR/../../build.py --build_dir /home/onnxruntimedev \
|
|
--config Debug Release \
|
|
--skip_submodule_sync --enable_onnx_tests \
|
|
--parallel --build_shared_lib \
|
|
--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
|
|
else
|
|
python3 $SCRIPT_DIR/../../build.py --build_dir /home/onnxruntimedev \
|
|
--config Debug Release --build_shared_lib \
|
|
--use_tvm \
|
|
--skip_submodule_sync --enable_onnx_tests \
|
|
--enable_pybind \
|
|
--parallel --use_openmp --use_mkldnn --build_shared_lib $BUILD_EXTR_PAR
|
|
/home/onnxruntimedev/Release/onnx_test_runner /data/onnx
|
|
fi
|