mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-01 23:30:35 +00:00
66980e4646
### Description Refactor the cmake code that is related to delay loading. Provide a cmake option to control if delay loading should be enabled or not. Disabling the option when python is enabled, due to a known issue. ### Motivation and Context ONNX Runtime's python package depends on DirectML.dll, but supposedly the DLL should be delay loaded. This PR only refactor the code. It doesn't change the behavior.
46 lines
1.5 KiB
Bash
Executable file
46 lines
1.5 KiB
Bash
Executable file
#!/bin/bash
|
|
set -ex
|
|
#Every cuda container has this $CUDA_VERSION env var set.
|
|
SHORT_CUDA_VERSION=$(echo $CUDA_VERSION | sed 's/\([[:digit:]]\+\.[[:digit:]]\+\)\.[[:digit:]]\+/\1/')
|
|
|
|
BUILD_ARGS=('--config' 'Release' '--update' '--build'
|
|
'--skip_submodule_sync'
|
|
'--build_shared_lib'
|
|
'--parallel' '--use_binskim_compliant_compile_flags'
|
|
'--build_wheel'
|
|
'--enable_onnx_tests'
|
|
'--use_cuda'
|
|
"--cuda_version=$SHORT_CUDA_VERSION"
|
|
"--cuda_home=/usr/local/cuda-$SHORT_CUDA_VERSION"
|
|
"--cudnn_home=/usr/local/cuda-$SHORT_CUDA_VERSION"
|
|
"--enable_cuda_profiling"
|
|
"--enable_cuda_nhwc_ops"
|
|
"--enable_pybind"
|
|
"--build_java"
|
|
"--cmake_extra_defines"
|
|
"CMAKE_CUDA_ARCHITECTURES=75"
|
|
"onnxruntime_BUILD_UNIT_TESTS=ON"
|
|
"onnxruntime_ENABLE_CUDA_EP_INTERNAL_TESTS=ON")
|
|
if [ -x "$(command -v ninja)" ]; then
|
|
BUILD_ARGS+=('--cmake_generator' 'Ninja')
|
|
fi
|
|
|
|
if [ -d /build ]; then
|
|
BUILD_ARGS+=('--build_dir' '/build')
|
|
else
|
|
BUILD_ARGS+=('--build_dir' 'build')
|
|
fi
|
|
|
|
if [ -x "$(command -v ccache)" ]; then
|
|
ccache -s;
|
|
#BUILD_ARGS+=("--use_cache")
|
|
fi
|
|
if [ -f /opt/python/cp312-cp312/bin/python3 ]; then
|
|
/opt/python/cp312-cp312/bin/python3 tools/ci_build/build.py "${BUILD_ARGS[@]}"
|
|
else
|
|
python3 tools/ci_build/build.py "${BUILD_ARGS[@]}"
|
|
fi
|
|
if [ -x "$(command -v ccache)" ]; then
|
|
ccache -sv
|
|
ccache -z
|
|
fi
|