diff --git a/CMakeLists.txt b/CMakeLists.txt index 8b2474401..efcaafc8e 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -79,30 +79,32 @@ if(ONNX_BUILD_PYTHON) target_link_libraries(onnx_opt_cpp2py_export PRIVATE onnx_optimizer) endif() -include(GNUInstallDirs) - -install(DIRECTORY ${PROJECT_SOURCE_DIR}/onnxoptimizer - DESTINATION ${CMAKE_INSTALL_INCLUDEDIR} - FILES_MATCHING - PATTERN "*.h") - -configure_file( - ${PROJECT_SOURCE_DIR}/cmake/ONNXOptimizerConfigVersion.cmake.in - ${PROJECT_BINARY_DIR}/ONNXOptimizerConfigVersion.cmake - @ONLY) -configure_file( - ${PROJECT_SOURCE_DIR}/cmake/ONNXOptimizerConfig.cmake.in - ${PROJECT_BINARY_DIR}/ONNXOptimizerConfig.cmake - @ONLY) -install(FILES - ${PROJECT_BINARY_DIR}/ONNXOptimizerConfigVersion.cmake - ${PROJECT_BINARY_DIR}/ONNXOptimizerConfig.cmake - DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/ONNXOptimizer - COMPONENT dev) -install(EXPORT ONNXOptimizerTargets DESTINATION "${CMAKE_INSTALL_LIBDIR}/cmake/ONNXOptimizer") -install(TARGETS - onnx_optimizer onnx_optimizer_c_api - EXPORT ONNXOptimizerTargets DESTINATION ${CMAKE_INSTALL_LIBDIR}) +if(ONNX_INSTALL) + include(GNUInstallDirs) + + install(DIRECTORY ${PROJECT_SOURCE_DIR}/onnxoptimizer + DESTINATION ${CMAKE_INSTALL_INCLUDEDIR} + FILES_MATCHING + PATTERN "*.h") + + configure_file( + ${PROJECT_SOURCE_DIR}/cmake/ONNXOptimizerConfigVersion.cmake.in + ${PROJECT_BINARY_DIR}/ONNXOptimizerConfigVersion.cmake + @ONLY) + configure_file( + ${PROJECT_SOURCE_DIR}/cmake/ONNXOptimizerConfig.cmake.in + ${PROJECT_BINARY_DIR}/ONNXOptimizerConfig.cmake + @ONLY) + install(FILES + ${PROJECT_BINARY_DIR}/ONNXOptimizerConfigVersion.cmake + ${PROJECT_BINARY_DIR}/ONNXOptimizerConfig.cmake + DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/ONNXOptimizer + COMPONENT dev) + install(EXPORT ONNXOptimizerTargets DESTINATION "${CMAKE_INSTALL_LIBDIR}/cmake/ONNXOptimizer") + install(TARGETS + onnx_optimizer onnx_optimizer_c_api + EXPORT ONNXOptimizerTargets DESTINATION ${CMAKE_INSTALL_LIBDIR}) +endif() if(ONNX_BUILD_TESTS) enable_testing() diff --git a/onnxoptimizer/passes/tensor_util.cc b/onnxoptimizer/passes/tensor_util.cc index 6f3849c2b..17c01eb3b 100644 --- a/onnxoptimizer/passes/tensor_util.cc +++ b/onnxoptimizer/passes/tensor_util.cc @@ -5,7 +5,7 @@ // ATTENTION: The code in this file is highly EXPERIMENTAL. // Adventurous users should note that the APIs will probably change. -#include +#include #include "onnx/common/platform_helpers.h" #include "onnxoptimizer/passes/tensor_util.h" diff --git a/setup.py b/setup.py index f48573709..eb7a9e472 100644 --- a/setup.py +++ b/setup.py @@ -163,6 +163,7 @@ def run(self): f"-DPython_EXECUTABLE={sys.executable}", "-DONNX_BUILD_PYTHON=ON", "-DCMAKE_EXPORT_COMPILE_COMMANDS=ON", + "-DONNX_INSTALL=OFF", f"-DONNX_NAMESPACE={ONNX_NAMESPACE}", "-DONNX_OPT_USE_SYSTEM_PROTOBUF={}".format( "ON" if ONNX_OPT_USE_SYSTEM_PROTOBUF else "OFF" diff --git a/tests/test_simple.cc b/tests/test_simple.cc index 649497d30..9572eeff9 100644 --- a/tests/test_simple.cc +++ b/tests/test_simple.cc @@ -20,7 +20,7 @@ TEST(OptimizerTest, NopReshape) { } )"; onnx::ModelProto model; - const onnx::Status status = onnx::OnnxParser::Parse(model, graph_str); + const onnx::Common::Status status = onnx::OnnxParser::Parse(model, graph_str); EXPECT_TRUE(status.IsOK()); auto optimized_model = onnx::optimization::Optimize(model, {"eliminate_nop_reshape", "eliminate_deadend"}); diff --git a/third_party/onnx b/third_party/onnx index 78925bf78..8847da48f 160000 --- a/third_party/onnx +++ b/third_party/onnx @@ -1 +1 @@ -Subproject commit 78925bf78b1bf61a3924b40bee652d4bcdba12f1 +Subproject commit 8847da48ff3bcf9314cdebb1a44114f723a8de4d