File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -12,6 +12,7 @@ export DNNL_FLAGS="--use_dnnl"
1212export CMAKE_ARGS=
1313export COREML_FLAGS=
1414export OPENVINO_FLAGS=
15+ export OPENVINO_CMAKE_DIR=
1516export OPENMP_FLAGS= # "--use_openmp"
1617export TRAINING_FLAGS= # --enable_training_apis --enable_training_ops
1718export CUDAFLAGS=" -v"
@@ -28,6 +29,7 @@ if [[ "$EXTENSION" == *gpu ]]; then
2829fi
2930
3031ONNXRUNTIME=1.26.0
32+ OPENVINO=2026.1.0
3133
3234mkdir -p " $PLATFORM$EXTENSION "
3335cd " $PLATFORM$EXTENSION "
@@ -72,13 +74,10 @@ case $PLATFORM in
7274esac
7375
7476case $PLATFORM in
75- linux-x86_64)
76- export OPENVINO_FLAGS=" --use_openvino CPU"
77- # Match OpenVINO's libstdc++ ABI to avoid protobuf/ONNX link errors.
78- export CXXFLAGS=" ${CXXFLAGS:- } -D_GLIBCXX_USE_CXX11_ABI=0"
79- ;;
80- windows-x86_64)
81- export OPENVINO_FLAGS=" --use_openvino CPU"
77+ linux-x86_64|windows-x86_64)
78+ " $PYTHON_BIN_PATH " -m pip install openvino==$OPENVINO
79+ OPENVINO_CMAKE_DIR=$( " $PYTHON_BIN_PATH " -c " import pathlib, openvino; print(pathlib.Path(openvino.__file__).parent / 'cmake')" )
80+ export OPENVINO_FLAGS=" --use_openvino CPU --cmake_extra_defines OpenVINO_DIR=$OPENVINO_CMAKE_DIR "
8281 ;;
8382esac
8483
You can’t perform that action at this time.
0 commit comments