fix ut

4 years ago · 470bd17be6
parent f14ab96a48
commit 470bd17be6
3 changed files with 8 additions and 2 deletions
--- a/tests/depends/error_manager/src/error_manager_stub.cc
+++ b/tests/depends/error_manager/src/error_manager_stub.cc
@ -18,6 +18,12 @@

 using namespace ErrorMessage;

+namespace ErrorMessage {
+int FormatErrorMessage(char *str_dst, size_t dst_max, const char *format, ...) {
+  return 1;
+}
+}
+
 thread_local Context ErrorManager::error_context_ = {0, "", "", ""};

  ErrorManager &ErrorManager::GetInstance() {
--- a/tests/ut/common/graph/CMakeLists.txt
+++ b/tests/ut/common/graph/CMakeLists.txt
@ -20,7 +20,7 @@ set(CMAKE_CXX_STANDARD 11)
 set(PROTO_LIST
    "${GE_CODE_DIR}/metadef/proto/om.proto"
    "${GE_CODE_DIR}/metadef/proto/ge_ir.proto"
-    "${GE_CODE_DIR}/metadef/proto/proto_inner/ge_onnx.proto"
+    "${GE_CODE_DIR}/metadef/proto/onnx/ge_onnx.proto"
 )

 protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})
--- a/tests/ut/ge/CMakeLists.txt
+++ b/tests/ut/ge/CMakeLists.txt
@ -33,7 +33,7 @@ set(PROTO_LIST
    "${GE_CODE_DIR}/metadef/proto/tensorflow/tensor_shape.proto"
    "${GE_CODE_DIR}/metadef/proto/tensorflow/types.proto"
    "${GE_CODE_DIR}/metadef/proto/tensorflow/node_def.proto"
-    "${GE_CODE_DIR}/metadef/proto/proto_inner/ge_onnx.proto"
+    "${GE_CODE_DIR}/metadef/proto/onnx/ge_onnx.proto"
    )

 protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})