Fix builds for ONNX on Windows

bytedeco · Oct 17, 2023 · 730e490 · 730e490
1 parent d347285
commit 730e490
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnx/cppbuild.sh b/onnx/cppbuild.sh
@@ -107,7 +107,7 @@ sedinplace 's/const std::string /std::string /g' onnx/defs/schema.h
 sedinplace 's/if WINDOWS:/if False:/g' setup.py
 sedinplace '/if platform.architecture/{N;N;N;d;}' setup.py
 sedinplace "/setup_requires.append('pytest-runner')/d" setup.py
-export CMAKE_ARGS="-DCMAKE_INSTALL_PREFIX=$INSTALL_PATH -DONNX_USE_PROTOBUF_SHARED_LIBS=OFF -DProtobuf_USE_STATIC_LIBS=ON -DONNX_USE_LITE_PROTO=ON"
+export CMAKE_ARGS="-DCMAKE_INSTALL_PREFIX=$INSTALL_PATH -DONNX_USE_PROTOBUF_SHARED_LIBS=OFF -DProtobuf_USE_STATIC_LIBS=ON -DONNX_USE_LITE_PROTO=ON -DPYTHON_LIBRARY=$PYTHON_LIB_PATH -DPYTHON_INCLUDE_DIR=$PYTHON_INCLUDE_PATH"
 $PYTHON_BIN_PATH -m pip install --target=$PYTHON_LIB_PATH setuptools==67.6.1 protobuf==$PROTO
 $PYTHON_BIN_PATH -m pip install -e .