diff options
Diffstat (limited to 'infra/nncc/cmake/packages/ONNXProtoConfig.cmake')
-rw-r--r-- | infra/nncc/cmake/packages/ONNXProtoConfig.cmake | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/infra/nncc/cmake/packages/ONNXProtoConfig.cmake b/infra/nncc/cmake/packages/ONNXProtoConfig.cmake new file mode 100644 index 000000000..a6dc7a01d --- /dev/null +++ b/infra/nncc/cmake/packages/ONNXProtoConfig.cmake @@ -0,0 +1,25 @@ +function(_ONNXProto_import) + nncc_find_package(ONNXSource EXACT 1.3.0 QUIET) + + if(NOT ONNXSource_FOUND) + set(ONNXProto_FOUND FALSE PARENT_SCOPE) + return() + endif(NOT ONNXSource_FOUND) + + nncc_find_package(Protobuf QUIET) + + if(NOT Protobuf_FOUND) + set(ONNXProto_FOUND FALSE PARENT_SCOPE) + return() + endif(NOT Protobuf_FOUND) + + if(NOT TARGET onnxproto) + nncc_include(ExternalProjectTools) + add_extdirectory("${CMAKE_CURRENT_LIST_DIR}/ONNXProto" onnxproto) + endif(NOT TARGET onnxproto) + + message(STATUS "Found ONNX: TRUE") + set(ONNXProto_FOUND TRUE PARENT_SCOPE) +endfunction(_ONNXProto_import) + +_ONNXProto_import() |