Browse Source

ge ut

tags/v1.1.0
t00456437 4 years ago
parent
commit
b911bd2a41
13 changed files with 602 additions and 511 deletions
  1. +8
    -1
      CMakeLists.txt
  2. +60
    -0
      cmake/external_libs/gtest.cmake
  3. +4
    -3
      cmake/external_libs/protobuf_shared.cmake
  4. +3
    -2
      cmake/external_libs/protoc.cmake
  5. +43
    -43
      tests/depends/cce/CMakeLists.txt
  6. +2
    -2
      tests/depends/mmpa/CMakeLists.txt
  7. +17
    -17
      tests/depends/omg/CMakeLists.txt
  8. +2
    -2
      tests/depends/slog/CMakeLists.txt
  9. +73
    -51
      tests/ut/common/graph/CMakeLists.txt
  10. +14
    -14
      tests/ut/common/graph/testcase/ge_graph/ge_format_refiner_unittest.cc
  11. +55
    -55
      tests/ut/common/graph/testcase/ge_graph/ge_model_serialize_unittest.cc
  12. +2
    -1
      tests/ut/common/graph/testcase/ge_graph/ge_node_unittest.cc
  13. +319
    -320
      tests/ut/ge/CMakeLists.txt

+ 8
- 1
CMakeLists.txt View File

@@ -30,6 +30,7 @@ if (ENABLE_OPEN_SRC)
include(cmake/external_libs/protobuf_static.cmake) include(cmake/external_libs/protobuf_static.cmake)
include(cmake/external_libs/protoc.cmake) include(cmake/external_libs/protoc.cmake)
include(cmake/external_libs/gflags.cmake) include(cmake/external_libs/gflags.cmake)
include(cmake/external_libs/gtest.cmake)
include(cmake/external_libs/securec.cmake) include(cmake/external_libs/securec.cmake)
include(cmake/external_libs/json.cmake) include(cmake/external_libs/json.cmake)
include(cmake/FindModule.cmake) include(cmake/FindModule.cmake)
@@ -71,6 +72,7 @@ if (ENABLE_OPEN_SRC)
else() else()
find_module(slog libslog.so ${ASCEND_ATC_DIR}) find_module(slog libslog.so ${ASCEND_ATC_DIR})
find_module(static_mmpa libmmpa.a ${ASCEND_ATC_DIR}) find_module(static_mmpa libmmpa.a ${ASCEND_ATC_DIR})
find_module(error_manager liberror_manager.so ${ASCEND_ATC_DIR})
if(PLATFORM STREQUAL "train") if(PLATFORM STREQUAL "train")
find_module(msprof libmsprof.so ${ASCEND_DRIVER_COMMON_DIR}) find_module(msprof libmsprof.so ${ASCEND_DRIVER_COMMON_DIR})
find_module(hccl libhccl.so ${ASCEND_RUNTIME_DIR}) find_module(hccl libhccl.so ${ASCEND_RUNTIME_DIR})
@@ -116,8 +118,13 @@ if (ENABLE_OPEN_SRC)
find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR}/driver) find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR}/driver)
#find_module(ascendcl_static libascendcl.a ${ASCEND_ACL_DIR}) #find_module(ascendcl_static libascendcl.a ${ASCEND_ACL_DIR})
else() else()
message(FATAL_ERROR "PLATFORM param is invalid, should be train or inference, build terminated")
message(STATUS "PLATFORM param is invalid, should be train or inference, you choose nothing!")
endif() endif()

if (ENABLE_GE_COV OR ENABLE_GE_UT OR ENABLE_GE_ST)
add_subdirectory(tests)
endif()

endif() endif()


set(METADEF_DIR ${CMAKE_CURRENT_LIST_DIR}/metadef) set(METADEF_DIR ${CMAKE_CURRENT_LIST_DIR}/metadef)


+ 60
- 0
cmake/external_libs/gtest.cmake View File

@@ -0,0 +1,60 @@
if (HAVE_GTEST)
return()
endif()

include(ExternalProject)

if ((${CMAKE_INSTALL_PREFIX} STREQUAL /usr/local) OR
(${CMAKE_INSTALL_PREFIX} STREQUAL "C:/Program Files (x86)/ascend"))
set(CMAKE_INSTALL_PREFIX ${GE_CODE_DIR}/output CACHE STRING "path for install()" FORCE)
message(STATUS "No install prefix selected, default to ${CMAKE_INSTALL_PREFIX}.")
endif()

if (ENABLE_GITEE)
set(REQ_URL "https://gitee.com/mirrors/googletest/repository/archive/release-1.8.0.tar.gz")
set(MD5 "")
else()
set(REQ_URL "https://github.com/google/googletest/archive/release-1.8.0.tar.gz")
set(MD5 "")
endif ()

set (gtest_CXXFLAGS "-D_GLIBCXX_USE_CXX11_ABI=0 -D_FORTIFY_SOURCE=2 -O2 -fstack-protector-all -Wl,-z,relro,-z,now,-z,noexecstack")
set (gtest_CFLAGS "-D_GLIBCXX_USE_CXX11_ABI=0 -D_FORTIFY_SOURCE=2 -O2 -fstack-protector-all -Wl,-z,relro,-z,now,-z,noexecstack")
ExternalProject_Add(gtest_build
URL ${REQ_URL}
CONFIGURE_COMMAND ${CMAKE_COMMAND} -DCMAKE_CXX_FLAGS=${gtest_CXXFLAGS} -DCMAKE_INSTALL_PREFIX=${CMAKE_INSTALL_PREFIX}/gtest <SOURCE_DIR>
-DBUILD_TESTING=OFF -DCMAKE_POSITION_INDEPENDENT_CODE=ON -DBUILD_SHARED_LIBS=ON -DCMAKE_MACOSX_RPATH=TRUE -Dgtest_disable_pthreads=ON
BUILD_COMMAND $(MAKE)
INSTALL_COMMAND $(MAKE) install
EXCLUDE_FROM_ALL TRUE
)

set(GTEST_PKG_DIR ${CMAKE_INSTALL_PREFIX}/gtest)

file(MAKE_DIRECTORY ${GTEST_PKG_DIR}/include)

add_library(gtest SHARED IMPORTED)

set_target_properties(gtest PROPERTIES
IMPORTED_LOCATION ${GTEST_PKG_DIR}/lib/libgtest.so
)

add_library(gtest_main SHARED IMPORTED)

set_target_properties(gtest_main PROPERTIES
IMPORTED_LOCATION ${GTEST_PKG_DIR}/lib/libgtest_main.so
)

target_include_directories(gtest INTERFACE ${GTEST_PKG_DIR}/include)
target_include_directories(gtest_main INTERFACE ${GTEST_PKG_DIR}/include)

set(INSTALL_BASE_DIR "")
set(INSTALL_LIBRARY_DIR lib)

install(FILES ${GTEST_PKG_DIR}/lib/libgtest.so ${GTEST_PKG_DIR}/lib/libgtest_main.so OPTIONAL
DESTINATION ${INSTALL_LIBRARY_DIR})

add_dependencies(gtest gtest_build)

#set(HAVE_GFLAGS TRUE CACHE BOOL "gflags build add")
set(HAVE_GTEST TRUE)

+ 4
- 3
cmake/external_libs/protobuf_shared.cmake View File

@@ -12,8 +12,9 @@ if ((${CMAKE_INSTALL_PREFIX} STREQUAL /usr/local) OR
endif() endif()


if (ENABLE_GITEE) if (ENABLE_GITEE)
set(REQ_URL "https://gitee.com/mirrors/protobuf_source/repository/archive/v3.8.0.tar.gz")
set(MD5 "eba86ae9f07ba5cfbaf8af3bc4e84236")
#set(REQ_URL "https://gitee.com/mirrors/protobuf_source/repository/archive/v3.8.0.tar.gz")
set(REQ_URL "/home/t00456437/workspace/v3.8.0.tar.gz")
#set(MD5 "eba86ae9f07ba5cfbaf8af3bc4e84236")
else() else()
set(REQ_URL "https://github.com/protocolbuffers/protobuf/archive/v3.8.0.tar.gz") set(REQ_URL "https://github.com/protocolbuffers/protobuf/archive/v3.8.0.tar.gz")
set(MD5 "3d9e32700639618a4d2d342c99d4507a") set(MD5 "3d9e32700639618a4d2d342c99d4507a")
@@ -22,7 +23,7 @@ endif ()
set(protobuf_CXXFLAGS "-Wno-maybe-uninitialized -Wno-unused-parameter -fPIC -fstack-protector-all -D_FORTIFY_SOURCE=2 -D_GLIBCXX_USE_CXX11_ABI=0 -O2 -Dgoogle=ascend_private") set(protobuf_CXXFLAGS "-Wno-maybe-uninitialized -Wno-unused-parameter -fPIC -fstack-protector-all -D_FORTIFY_SOURCE=2 -D_GLIBCXX_USE_CXX11_ABI=0 -O2 -Dgoogle=ascend_private")
set(protobuf_LDFLAGS "-Wl,-z,relro,-z,now,-z,noexecstack") set(protobuf_LDFLAGS "-Wl,-z,relro,-z,now,-z,noexecstack")
ExternalProject_Add(protobuf_build ExternalProject_Add(protobuf_build
URL https://github.com/protocolbuffers/protobuf/archive/v3.8.0.tar.gz
URL ${REQ_URL}
CONFIGURE_COMMAND ${CMAKE_COMMAND} CONFIGURE_COMMAND ${CMAKE_COMMAND}
-Dprotobuf_WITH_ZLIB=OFF -Dprotobuf_WITH_ZLIB=OFF
-DCMAKE_INSTALL_LIBDIR=${CMAKE_INSTALL_LIBDIR} -DCMAKE_INSTALL_LIBDIR=${CMAKE_INSTALL_LIBDIR}


+ 3
- 2
cmake/external_libs/protoc.cmake View File

@@ -13,8 +13,9 @@ if ((${CMAKE_INSTALL_PREFIX} STREQUAL /usr/local) OR
endif() endif()
if (ENABLE_GITEE) if (ENABLE_GITEE)
set(REQ_URL "https://gitee.com/mirrors/protobuf_source/repository/archive/v3.8.0.tar.gz")
set(MD5 "eba86ae9f07ba5cfbaf8af3bc4e84236")
#set(REQ_URL "https://gitee.com/mirrors/protobuf_source/repository/archive/v3.8.0.tar.gz")
#set(MD5 "eba86ae9f07ba5cfbaf8af3bc4e84236")
set(REQ_URL "/home/t00456437/workspace/v3.8.0.tar.gz")
else() else()
set(REQ_URL "https://github.com/protocolbuffers/protobuf/archive/v3.8.0.tar.gz") set(REQ_URL "https://github.com/protocolbuffers/protobuf/archive/v3.8.0.tar.gz")
set(MD5 "3d9e32700639618a4d2d342c99d4507a") set(MD5 "3d9e32700639618a4d2d342c99d4507a")


+ 43
- 43
tests/depends/cce/CMakeLists.txt View File

@@ -13,60 +13,60 @@
# limitations under the License. # limitations under the License.
# ============================================================================ # ============================================================================


cmake_minimum_required(VERSION 2.8)
#cmake_minimum_required(VERSION 2.8)


project(STUB_CCE) project(STUB_CCE)


set(CMAKE_CXX_STANDARD 11) set(CMAKE_CXX_STANDARD 11)


include_directories(${GE_SOURCE_DIR}/inc)
include_directories(${GE_SOURCE_DIR}/inc/framework)
include_directories(${GE_SOURCE_DIR}/inc/graph)
include_directories(${GE_SOURCE_DIR}/inc/external)
include_directories(${GE_SOURCE_DIR}/inc/external/graph)
include_directories(${GE_SOURCE_DIR}/src/common)
include_directories(${GE_SOURCE_DIR}/src/common/graph)
include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc)
include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc/cce)
include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc/ops)
include_directories(${GE_CODE_DIR}/inc)
include_directories(${GE_CODE_DIR}/inc/framework)
include_directories(${GE_CODE_DIR}/metadef/inc/graph)
include_directories(${GE_CODE_DIR}/inc/external)
include_directories(${GE_CODE_DIR}/metadef/inc/external/graph)
include_directories(${GE_CODE_DIR}/metadef)
include_directories(${GE_CODE_DIR}/metadef/graph)
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc)
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/cce)
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/ops)
include_directories(${CMAKE_BINARY_DIR}) include_directories(${CMAKE_BINARY_DIR})
include_directories(${CMAKE_BINARY_DIR}/proto/ge) include_directories(${CMAKE_BINARY_DIR}/proto/ge)
file(GLOB_RECURSE PROTO_LIST RELATIVE ${CMAKE_CURRENT_LIST_DIR}
"${GE_SOURCE_DIR}/src/proto/om.proto"
"${GE_SOURCE_DIR}/src/proto/ge_ir.proto"
"${GE_SOURCE_DIR}/src/proto/task.proto"
set(PROTO_LIST
"${GE_CODE_DIR}/metadef/proto/om.proto"
"${GE_CODE_DIR}/metadef/proto/ge_ir.proto"
"${GE_CODE_DIR}/metadef/proto/task.proto"
) )


ge_protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})
protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})


file(GLOB_RECURSE SRCS RELATIVE ${CMAKE_CURRENT_LIST_DIR}
"${GE_SOURCE_DIR}/src/common/graph/ge_attr_define.cc"
"${GE_SOURCE_DIR}/src/common/graph/anchor.cc"
"${GE_SOURCE_DIR}/src/common/graph/ge_attr_value.cc"
"${GE_SOURCE_DIR}/src/common/graph/buffer.cc"
"${GE_SOURCE_DIR}/src/common/graph/compute_graph.cc"
"${GE_SOURCE_DIR}/src/common/graph/graph.cc"
"${GE_SOURCE_DIR}/src/common/graph/model.cc"
"${GE_SOURCE_DIR}/src/common/graph/model_serialize.cc"
"${GE_SOURCE_DIR}/src/common/graph/node.cc"
"${GE_SOURCE_DIR}/src/common/graph/op_desc.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator_factory.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator_factory_impl.cc"
"${GE_SOURCE_DIR}/src/common/graph/tensor.cc"
"${GE_SOURCE_DIR}/src/common/graph/detail/attributes_holder.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/anchor_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/graph_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/node_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/op_desc_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/type_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/op_imp.cc"
"${GE_SOURCE_DIR}/src/common/graph/shape_refiner.cc"
"${GE_SOURCE_DIR}/src/common/graph/ge_tensor.cc"
"${GE_SOURCE_DIR}/src/common/graph/opsproto/opsproto_manager.cc"
set(SRCS
"${GE_CODE_DIR}/metadef/graph/ge_attr_define.cc"
"${GE_CODE_DIR}/metadef/graph/anchor.cc"
"${GE_CODE_DIR}/metadef/graph/ge_attr_value.cc"
"${GE_CODE_DIR}/metadef/graph/buffer.cc"
"${GE_CODE_DIR}/metadef/graph/compute_graph.cc"
"${GE_CODE_DIR}/metadef/graph/graph.cc"
"${GE_CODE_DIR}/metadef/graph/model.cc"
"${GE_CODE_DIR}/metadef/graph/model_serialize.cc"
"${GE_CODE_DIR}/metadef/graph/node.cc"
"${GE_CODE_DIR}/metadef/graph/op_desc.cc"
"${GE_CODE_DIR}/metadef/graph/operator.cc"
"${GE_CODE_DIR}/metadef/graph/operator_factory.cc"
"${GE_CODE_DIR}/metadef/graph/operator_factory_impl.cc"
"${GE_CODE_DIR}/metadef/graph/tensor.cc"
"${GE_CODE_DIR}/metadef/graph/detail/attributes_holder.cc"
"${GE_CODE_DIR}/metadef/graph/utils/anchor_utils.cc"
"${GE_CODE_DIR}/metadef/graph/utils/graph_utils.cc"
"${GE_CODE_DIR}/metadef/graph/utils/node_utils.cc"
"${GE_CODE_DIR}/metadef/graph/utils/op_desc_utils.cc"
"${GE_CODE_DIR}/metadef/graph/utils/type_utils.cc"
"${GE_CODE_DIR}/metadef/ops/op_imp.cpp"
"${GE_CODE_DIR}/metadef/graph/shape_refiner.cc"
"${GE_CODE_DIR}/metadef/graph/ge_tensor.cc"
"${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc"
) )
add_library(cce_ge_stub SHARED src/cce_stub.cc ${PROTO_SRCS} ${PROTO_HDRS}) add_library(cce_ge_stub SHARED src/cce_stub.cc ${PROTO_SRCS} ${PROTO_HDRS})
target_link_libraries(cce_ge_stub protobuf::protobuf)
target_link_libraries(cce_ge_stub ascend_protobuf)


add_library(cce_stub SHARED ${SRCS} ${PROTO_SRCS} ${PROTO_HDRS}) add_library(cce_stub SHARED ${SRCS} ${PROTO_SRCS} ${PROTO_HDRS})
target_link_libraries(cce_stub protobuf::protobuf)
target_link_libraries(cce_stub ascend_protobuf)

+ 2
- 2
tests/depends/mmpa/CMakeLists.txt View File

@@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
# ============================================================================ # ============================================================================


cmake_minimum_required(VERSION 2.8)
#cmake_minimum_required(VERSION 2.8)


project(STUB_MMPA) project(STUB_MMPA)


@@ -27,4 +27,4 @@ include_directories(${GE_SOURCE_DIR}/inc/framework)
include_directories(${GE_SOURCE_DIR}/inc/external) include_directories(${GE_SOURCE_DIR}/inc/external)


add_library(mmpa_stub SHARED ${SRCS}) add_library(mmpa_stub SHARED ${SRCS})
target_link_libraries(mmpa_stub protobuf::protobuf)
target_link_libraries(mmpa_stub ascend_protobuf)

+ 17
- 17
tests/depends/omg/CMakeLists.txt View File

@@ -13,33 +13,33 @@
# limitations under the License. # limitations under the License.
# ============================================================================ # ============================================================================


cmake_minimum_required(VERSION 2.8)
#cmake_minimum_required(VERSION 2.8)


project(OMG_CCE) project(OMG_CCE)


set(CMAKE_CXX_STANDARD 11) set(CMAKE_CXX_STANDARD 11)


include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc)
include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc/cce)
include_directories(${GE_SOURCE_DIR}/inc)
include_directories(${GE_SOURCE_DIR}/inc/framework)
include_directories(${GE_SOURCE_DIR}/inc/graph)
include_directories(${GE_SOURCE_DIR}/inc/external)
include_directories(${GE_SOURCE_DIR}/inc/external/graph)
include_directories(${GE_SOURCE_DIR}/src/ge)
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc)
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/cce)
include_directories(${GE_CODE_DIR}/inc)
include_directories(${GE_CODE_DIR}/inc/framework)
include_directories(${GE_CODE_DIR}/metadef/inc/graph)
include_directories(${GE_CODE_DIR}/inc/external)
include_directories(${GE_CODE_DIR}/metadef/inc/external/graph)
include_directories(${GE_CODE_DIR}/ge)
include_directories(${CMAKE_BINARY_DIR}) include_directories(${CMAKE_BINARY_DIR})
include_directories(${CMAKE_BINARY_DIR}/proto/ge) include_directories(${CMAKE_BINARY_DIR}/proto/ge)
file(GLOB_RECURSE PROTO_LIST RELATIVE ${CMAKE_CURRENT_LIST_DIR}
"${GE_SOURCE_DIR}/src/proto/om.proto"
"${GE_SOURCE_DIR}/src/proto/task.proto"
set(PROTO_LIST
"${GE_CODE_DIR}/metadef/proto/om.proto"
"${GE_CODE_DIR}/metadef/proto/task.proto"
) )


ge_protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})
protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})


file(GLOB_RECURSE SRCS RELATIVE ${CMAKE_CURRENT_LIST_DIR}
# "${GE_SOURCE_DIR}/src/ge/common/util.cc"
"src/omg_stub.cc"
set(SRCS
# "${GE_CODE_DIR}/src/ge/common/util.cc"
"src/omg_stub.cc"
) )


add_library(omg_stub SHARED ${SRCS} ${PROTO_SRCS} ${PROTO_HDRS}) add_library(omg_stub SHARED ${SRCS} ${PROTO_SRCS} ${PROTO_HDRS})
target_link_libraries(omg_stub protobuf::protobuf)
target_link_libraries(omg_stub ascend_protobuf)

+ 2
- 2
tests/depends/slog/CMakeLists.txt View File

@@ -19,5 +19,5 @@ project(slog_stub)
file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
"src/*.cc" "src/*.cc"
) )
include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc)
add_library(slog_stub SHARED ${SRC_FILES})
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc)
add_library(slog_stub SHARED ${SRC_FILES})

+ 73
- 51
tests/ut/common/graph/CMakeLists.txt View File

@@ -17,30 +17,34 @@ project(ut_libgraph)


set(CMAKE_CXX_STANDARD 11) set(CMAKE_CXX_STANDARD 11)


file(GLOB_RECURSE PROTO_LIST RELATIVE ${CMAKE_CURRENT_LIST_DIR}
"${GE_SOURCE_DIR}/src/proto/om.proto"
"${GE_SOURCE_DIR}/src/proto/ge_ir.proto"
"${onnx_INC}/onnx/onnx.proto"
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_protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})
protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})


# include directories # include directories
include_directories(${CMAKE_CURRENT_LIST_DIR}) include_directories(${CMAKE_CURRENT_LIST_DIR})
include_directories(${GE_SOURCE_DIR}/src)
include_directories(${GE_SOURCE_DIR}/src/common)
include_directories(${GE_SOURCE_DIR}/src/common/graph)
include_directories(${GE_SOURCE_DIR}/inc)
include_directories(${GE_SOURCE_DIR}/inc/external)
include_directories(${GE_SOURCE_DIR}/inc/external/graph)
include_directories(${GE_SOURCE_DIR}/inc/graph)
include_directories(${GE_SOURCE_DIR}/inc/common)
include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc)
include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc/ops)
include_directories(${GE_CODE_DIR})
include_directories(${GE_CODE_DIR}/metadef)
include_directories(${GE_CODE_DIR}/metadef/graph)
include_directories(${GE_CODE_DIR}/inc)
include_directories(${GE_CODE_DIR}/inc/external)
include_directories(${GE_CODE_DIR}/metadef/inc/external)
include_directories(${GE_CODE_DIR}/metadef/inc/external/graph)
include_directories(${GE_CODE_DIR}/metadef/inc)
include_directories(${GE_CODE_DIR}/metadef/inc/graph)
include_directories(${GE_CODE_DIR}/metadef/inc/common)
include_directories(${GE_CODE_DIR}/metadef/third_party)
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc)
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/ops)
include_directories(${CMAKE_BINARY_DIR}) include_directories(${CMAKE_BINARY_DIR})
include_directories(${CMAKE_BINARY_DIR}/proto/ge) include_directories(${CMAKE_BINARY_DIR}/proto/ge)
include_directories(${CMAKE_BINARY_DIR}/proto/ge/proto)


file(GLOB_RECURSE UT_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
set(UT_FILES
"testcase/ge_graph/ge_anchor_utils_unittest.cc" "testcase/ge_graph/ge_anchor_utils_unittest.cc"
"testcase/ge_graph/ge_def_type_unittest.cc" "testcase/ge_graph/ge_def_type_unittest.cc"
"testcase/ge_graph/ge_graph_anchor_unittest.cc" "testcase/ge_graph/ge_graph_anchor_unittest.cc"
@@ -56,41 +60,59 @@ file(GLOB_RECURSE UT_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
"testcase/ge_graph/ge_model_unittest.cc" "testcase/ge_graph/ge_model_unittest.cc"
) )


file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/common/graph/option/ge_local_context.cc"
"${GE_SOURCE_DIR}/src/common/graph/option/ge_context.cc"
"${GE_SOURCE_DIR}/src/common/graph/anchor.cc"
"${GE_SOURCE_DIR}/src/common/graph/ge_attr_value.cc"
"${GE_SOURCE_DIR}/src/common/graph/attr_value.cc"
"${GE_SOURCE_DIR}/src/common/graph/buffer.cc"
"${GE_SOURCE_DIR}/src/common/graph/compute_graph.cc"
"${GE_SOURCE_DIR}/src/common/graph/ge_attr_define.cc"
"${GE_SOURCE_DIR}/src/common/graph/graph.cc"
"${GE_SOURCE_DIR}/src/common/graph/model.cc"
"${GE_SOURCE_DIR}/src/common/graph/model_serialize.cc"
"${GE_SOURCE_DIR}/src/common/graph/node.cc"
"${GE_SOURCE_DIR}/src/common/graph/op_desc.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator_reg.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator_factory.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator_factory_impl.cc"
"${GE_SOURCE_DIR}/src/common/graph/range_vistor.cc"
"${GE_SOURCE_DIR}/src/common/graph/tensor.cc"
"${GE_SOURCE_DIR}/src/common/graph/ge_tensor.cc"
"${GE_SOURCE_DIR}/src/common/graph/shape_refiner.cc"
"${GE_SOURCE_DIR}/src/common/graph/format_refiner.cc"
"${GE_SOURCE_DIR}/src/common/graph/inference_context.cc"
"${GE_SOURCE_DIR}/src/common/graph/detail/attributes_holder.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/anchor_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/graph_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/node_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/op_desc_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/type_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/ge_ir_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/tensor_utils.cc"
"${GE_SOURCE_DIR}/src/common/ops/op_imp.cc"
"${GE_SOURCE_DIR}/src/common/graph/opsproto/opsproto_manager.cc"
set(SRC_FILES
#"${GE_CODE_DIR}/metadef/graph/option/ge_local_context.cc"
#"${GE_CODE_DIR}/metadef/graph/option/ge_context.cc"
#"${GE_CODE_DIR}/metadef/graph/anchor.cc"
#"${GE_CODE_DIR}/metadef/graph/ge_attr_value.cc"
#"${GE_CODE_DIR}/metadef/graph/attr_value.cc"
#"${GE_CODE_DIR}/metadef/graph/buffer.cc"
#"${GE_CODE_DIR}/metadef/graph/compute_graph.cc"
#"${GE_CODE_DIR}/metadef/graph/ge_attr_define.cc"
#"${GE_CODE_DIR}/metadef/graph/graph.cc"
#"${GE_CODE_DIR}/metadef/graph/gnode.cc"
#"${GE_CODE_DIR}/metadef/graph/ascend_string.cc"
#"${GE_CODE_DIR}/metadef/graph/model.cc"
#"${GE_CODE_DIR}/metadef/graph/model_serialize.cc"
#"${GE_CODE_DIR}/metadef/graph/node.cc"
#"${GE_CODE_DIR}/metadef/graph/op_desc.cc"
#"${GE_CODE_DIR}/metadef/graph/operator.cc"
#"${GE_CODE_DIR}/metadef/graph/operator_reg.cc"
#"${GE_CODE_DIR}/metadef/graph/operator_factory.cc"
#"${GE_CODE_DIR}/metadef/graph/operator_factory_impl.cc"
#"${GE_CODE_DIR}/metadef/graph/range_vistor.cc"
#"${GE_CODE_DIR}/metadef/graph/tensor.cc"
#"${GE_CODE_DIR}/metadef/graph/ge_tensor.cc"
#"${GE_CODE_DIR}/metadef/graph/shape_refiner.cc"
#"${GE_CODE_DIR}/metadef/graph/format_refiner.cc"
#"${GE_CODE_DIR}/metadef/graph/inference_context.cc"
#"${GE_CODE_DIR}/metadef/graph/detail/attributes_holder.cc"
#"${GE_CODE_DIR}/metadef/graph/utils/anchor_utils.cc"
#"${GE_CODE_DIR}/metadef/graph/utils/graph_utils.cc"
#"${GE_CODE_DIR}/metadef/graph/utils/node_utils.cc"
#"${GE_CODE_DIR}/metadef/graph/utils/op_desc_utils.cc"
#"${GE_CODE_DIR}/metadef/graph/utils/type_utils.cc"
#"${GE_CODE_DIR}/metadef/graph/utils/ge_ir_utils.cc"
#"${GE_CODE_DIR}/metadef/graph/utils/tensor_utils.cc"
"${GE_CODE_DIR}/metadef/ops/op_imp.cpp"
#"${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc"
) )


#add_executable(ut_libgraph ${UT_FILES} ${SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
add_executable(ut_libgraph ${UT_FILES} ${SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) add_executable(ut_libgraph ${UT_FILES} ${SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
target_link_libraries(ut_libgraph graphengine::gtest graphengine::gtest_main slog_stub protobuf::protobuf graphengine::securec rt dl)

target_compile_definitions(ut_libgraph PRIVATE
google=ascend_private
)

target_link_libraries(ut_libgraph
$<BUILD_INTERFACE:intf_pub>
graph
gtest
gtest_main
slog_stub
ascend_protobuf
c_sec
-lrt
-ldl
)

+ 14
- 14
tests/ut/common/graph/testcase/ge_graph/ge_format_refiner_unittest.cc View File

@@ -85,7 +85,7 @@ ut::GraphBuilder BuildGraph1() {
builder.AddDataEdge(var2, 0, conv1, 1); builder.AddDataEdge(var2, 0, conv1, 1);
builder.AddDataEdge(conv1, 0, relu1, 0); builder.AddDataEdge(conv1, 0, relu1, 0);
builder.AddDataEdge(relu1, 0, netoutput1, 0); builder.AddDataEdge(relu1, 0, netoutput1, 0);
FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);
return builder; return builder;
} }


@@ -134,7 +134,7 @@ ut::GraphBuilder BuildGraph2() {
builder.AddDataEdge(var6, 0, bn1, 4); builder.AddDataEdge(var6, 0, bn1, 4);
builder.AddDataEdge(bn1, 0, relu1, 0); builder.AddDataEdge(bn1, 0, relu1, 0);
builder.AddDataEdge(relu1, 0, netoutput1, 0); builder.AddDataEdge(relu1, 0, netoutput1, 0);
FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);
return builder; return builder;
} }


@@ -189,7 +189,7 @@ ut::GraphBuilder BuildGraph3() {
builder.AddDataEdge(relu1, 0, conv2, 0); builder.AddDataEdge(relu1, 0, conv2, 0);
builder.AddDataEdge(var3, 0, conv2, 1); builder.AddDataEdge(var3, 0, conv2, 1);
builder.AddDataEdge(conv2, 0, netoutput1, 0); builder.AddDataEdge(conv2, 0, netoutput1, 0);
FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);
return builder; return builder;
} }


@@ -248,7 +248,7 @@ ut::GraphBuilder BuildGraph4() {
builder.AddDataEdge(relu1, 0, conv2, 0); builder.AddDataEdge(relu1, 0, conv2, 0);
builder.AddDataEdge(var3, 0, conv2, 1); builder.AddDataEdge(var3, 0, conv2, 1);
builder.AddDataEdge(conv2, 0, netoutput1, 0); builder.AddDataEdge(conv2, 0, netoutput1, 0);
FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);
return builder; return builder;
} }


@@ -305,7 +305,7 @@ ut::GraphBuilder BuilderGraph5() {
builder.AddDataEdge(relug1, 0, bng1, 0); builder.AddDataEdge(relug1, 0, bng1, 0);
builder.AddDataEdge(bng1, 0, apply1, 0); builder.AddDataEdge(bng1, 0, apply1, 0);
builder.AddDataEdge(apply1, 0, netoutput1, 0); builder.AddDataEdge(apply1, 0, netoutput1, 0);
FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);
return builder; return builder;
} }


@@ -353,7 +353,7 @@ ut::GraphBuilder BuildGraph6() {
builder.AddDataEdge(constant, 0, addn, 2); builder.AddDataEdge(constant, 0, addn, 2);
builder.AddDataEdge(addn, 0, netoutput, 0); builder.AddDataEdge(addn, 0, netoutput, 0);


FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);


return builder; return builder;
} }
@@ -397,7 +397,7 @@ ut::GraphBuilder BuildGraph7() {
builder.AddDataEdge(constant, 0, addn, 2); builder.AddDataEdge(constant, 0, addn, 2);
builder.AddDataEdge(addn, 0, netoutput, 0); builder.AddDataEdge(addn, 0, netoutput, 0);


FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);


return builder; return builder;
} }
@@ -449,7 +449,7 @@ ut::GraphBuilder BuildGraph8() {
builder.AddDataEdge(relu, 0, reshape, 0); builder.AddDataEdge(relu, 0, reshape, 0);
builder.AddDataEdge(reshape, 0, conv, 1); builder.AddDataEdge(reshape, 0, conv, 1);
builder.AddDataEdge(conv, 0, netoutput, 0); builder.AddDataEdge(conv, 0, netoutput, 0);
FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);
return builder; return builder;
} }
} // namespace } // namespace
@@ -457,7 +457,7 @@ ut::GraphBuilder BuildGraph8() {
TEST_F(UtestFormatRefiner, data_format) { TEST_F(UtestFormatRefiner, data_format) {
auto builder = BuildGraph8(); auto builder = BuildGraph8();
auto graph = builder.GetGraph(); auto graph = builder.GetGraph();
FormatRefiner::SetInferOrigineFormatFlag(false);
//FormatRefiner::SetInferOrigineFormatFlag(false);
graph->SaveDataFormat(FORMAT_NCHW); graph->SaveDataFormat(FORMAT_NCHW);
EXPECT_EQ(FormatRefiner::InferOrigineFormat(graph), GRAPH_SUCCESS); EXPECT_EQ(FormatRefiner::InferOrigineFormat(graph), GRAPH_SUCCESS);
auto data2 = graph->FindNode("data2"); auto data2 = graph->FindNode("data2");
@@ -466,18 +466,18 @@ TEST_F(UtestFormatRefiner, data_format) {
EXPECT_EQ(data2->GetOpDesc()->GetOutputDesc(0).GetOriginFormat(), FORMAT_NCHW); EXPECT_EQ(data2->GetOpDesc()->GetOutputDesc(0).GetOriginFormat(), FORMAT_NCHW);
EXPECT_EQ(relu->GetOpDesc()->GetInputDesc(0).GetOriginFormat(), FORMAT_NCHW); EXPECT_EQ(relu->GetOpDesc()->GetInputDesc(0).GetOriginFormat(), FORMAT_NCHW);
EXPECT_EQ(relu->GetOpDesc()->GetOutputDesc(0).GetOriginFormat(), FORMAT_NCHW); EXPECT_EQ(relu->GetOpDesc()->GetOutputDesc(0).GetOriginFormat(), FORMAT_NCHW);
FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);
} }


TEST_F(UtestFormatRefiner, constant_fail) { TEST_F(UtestFormatRefiner, constant_fail) {
FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);
auto builder = BuildGraph6(); auto builder = BuildGraph6();
auto graph = builder.GetGraph(); auto graph = builder.GetGraph();
EXPECT_EQ(FormatRefiner::InferOrigineFormat(graph), GRAPH_FAILED); EXPECT_EQ(FormatRefiner::InferOrigineFormat(graph), GRAPH_FAILED);
} }


TEST_F(UtestFormatRefiner, scalar_nodes_infer) { TEST_F(UtestFormatRefiner, scalar_nodes_infer) {
FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);
auto builder = BuildGraph6(); auto builder = BuildGraph6();
auto graph = builder.GetGraph(); auto graph = builder.GetGraph();
auto constant = graph->FindNode("constant"); auto constant = graph->FindNode("constant");
@@ -650,7 +650,7 @@ TEST_F(UtestFormatRefiner, infer_origine_format_failed) {
} }


TEST_F(UtestFormatRefiner, save_format) { TEST_F(UtestFormatRefiner, save_format) {
FormatRefiner::SetInferOrigineFormatFlag(true);
//FormatRefiner::SetInferOrigineFormatFlag(true);
auto builder = BuildGraph6(); auto builder = BuildGraph6();
auto graph = builder.GetGraph(); auto graph = builder.GetGraph();
graph->SaveDataFormat(FORMAT_NHWC); graph->SaveDataFormat(FORMAT_NHWC);
@@ -658,4 +658,4 @@ TEST_F(UtestFormatRefiner, save_format) {
EXPECT_EQ(save_format, FORMAT_NHWC); EXPECT_EQ(save_format, FORMAT_NHWC);
graph->SaveDataFormat(FORMAT_ND); graph->SaveDataFormat(FORMAT_ND);
} }
} // namespace ge
} // namespace ge

+ 55
- 55
tests/ut/common/graph/testcase/ge_graph/ge_model_serialize_unittest.cc View File

@@ -1060,7 +1060,7 @@ TEST(UtestGeModelSerialize, test_model_serialize_imp_invalid_param) {
auto graph = std::make_shared<ComputeGraph>("test_graph"); auto graph = std::make_shared<ComputeGraph>("test_graph");
auto node = graph->AddNode(std::make_shared<OpDesc>()); auto node = graph->AddNode(std::make_shared<OpDesc>());
node->op_ = nullptr; node->op_ = nullptr;
proto::ModelDef model_def;
ge::proto::ModelDef model_def;
Model model; Model model;
model.SetGraph(GraphUtils::CreateGraphFromComputeGraph(graph)); model.SetGraph(GraphUtils::CreateGraphFromComputeGraph(graph));
EXPECT_FALSE(imp.SerializeModel(model, &model_def)); EXPECT_FALSE(imp.SerializeModel(model, &model_def));
@@ -1101,26 +1101,26 @@ TEST(UTEST_ge_model_unserialize, test_invalid_tensor) {


TEST(UTEST_ge_model_unserialize, test_invalid_TensorDesc) { TEST(UTEST_ge_model_unserialize, test_invalid_TensorDesc) {
{ // valid { // valid
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.mutable_attr(); auto attrs = mode_def.mutable_attr();


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
auto tensor_desc_attr = attr_def->mutable_td(); auto tensor_desc_attr = attr_def->mutable_td();
tensor_desc_attr->set_layout("NCHW"); tensor_desc_attr->set_layout("NCHW");
tensor_desc_attr->set_dtype(proto::DataType::DT_INT8);
tensor_desc_attr->set_dtype(ge::proto::DataType::DT_INT8);


ModelSerializeImp imp; ModelSerializeImp imp;
Model model; Model model;
EXPECT_TRUE(imp.UnserializeModel(model, mode_def)); EXPECT_TRUE(imp.UnserializeModel(model, mode_def));
} }
{ // invalid layout { // invalid layout
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.mutable_attr(); auto attrs = mode_def.mutable_attr();


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
auto tensor_desc_attr = attr_def->mutable_td(); auto tensor_desc_attr = attr_def->mutable_td();
tensor_desc_attr->set_layout("InvalidLayout"); tensor_desc_attr->set_layout("InvalidLayout");
tensor_desc_attr->set_dtype(proto::DataType::DT_INT8);
tensor_desc_attr->set_dtype(ge::proto::DataType::DT_INT8);


ModelSerializeImp imp; ModelSerializeImp imp;
Model model; Model model;
@@ -1131,13 +1131,13 @@ TEST(UTEST_ge_model_unserialize, test_invalid_TensorDesc) {
EXPECT_EQ(tensor_desc.GetDataType(), DT_INT8); EXPECT_EQ(tensor_desc.GetDataType(), DT_INT8);
} }
{ // invalid datatype { // invalid datatype
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.mutable_attr(); auto attrs = mode_def.mutable_attr();


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
auto tensor_desc_attr = attr_def->mutable_td(); // tensor desc auto tensor_desc_attr = attr_def->mutable_td(); // tensor desc
tensor_desc_attr->set_layout("NHWC"); tensor_desc_attr->set_layout("NHWC");
tensor_desc_attr->set_dtype((proto::DataType)100);
tensor_desc_attr->set_dtype((ge::proto::DataType)100);


ModelSerializeImp imp; ModelSerializeImp imp;
Model model; Model model;
@@ -1148,13 +1148,13 @@ TEST(UTEST_ge_model_unserialize, test_invalid_TensorDesc) {
EXPECT_EQ(tensor_desc.GetDataType(), DT_UNDEFINED); EXPECT_EQ(tensor_desc.GetDataType(), DT_UNDEFINED);
} }
{ // invalid datatype { // invalid datatype
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.mutable_attr(); auto attrs = mode_def.mutable_attr();


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
auto tensor_desc_attr = attr_def->mutable_t()->mutable_desc(); // tensor auto tensor_desc_attr = attr_def->mutable_t()->mutable_desc(); // tensor
tensor_desc_attr->set_layout("NHWC"); tensor_desc_attr->set_layout("NHWC");
tensor_desc_attr->set_dtype((proto::DataType)100);
tensor_desc_attr->set_dtype((ge::proto::DataType)100);


ModelSerializeImp imp; ModelSerializeImp imp;
Model model; Model model;
@@ -1167,13 +1167,13 @@ TEST(UTEST_ge_model_unserialize, test_invalid_TensorDesc) {
EXPECT_EQ(tensor_desc.GetDataType(), DT_UNDEFINED); EXPECT_EQ(tensor_desc.GetDataType(), DT_UNDEFINED);
} }
{ // invalid attrmap { // invalid attrmap
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.add_graph()->mutable_attr(); // graph attr auto attrs = mode_def.add_graph()->mutable_attr(); // graph attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
auto tensor_desc_attr = attr_def->mutable_t()->mutable_desc(); // tensor auto tensor_desc_attr = attr_def->mutable_t()->mutable_desc(); // tensor
tensor_desc_attr->set_layout("NCHW"); tensor_desc_attr->set_layout("NCHW");
tensor_desc_attr->set_dtype(proto::DataType::DT_INT8);
tensor_desc_attr->set_dtype(ge::proto::DataType::DT_INT8);
auto attrs1 = tensor_desc_attr->mutable_attr(); auto attrs1 = tensor_desc_attr->mutable_attr();
auto attr1 = (*attrs1)["key2"]; // empty attr auto attr1 = (*attrs1)["key2"]; // empty attr


@@ -1191,13 +1191,13 @@ TEST(UTEST_ge_model_unserialize, test_invalid_TensorDesc) {
EXPECT_EQ(attr_value.GetValueType(), GeAttrValue::VT_NONE); EXPECT_EQ(attr_value.GetValueType(), GeAttrValue::VT_NONE);
} }
{ // invalid attrmap2 { // invalid attrmap2
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
auto tensor_desc_attr = attr_def->mutable_t()->mutable_desc(); // tensor auto tensor_desc_attr = attr_def->mutable_t()->mutable_desc(); // tensor
tensor_desc_attr->set_layout("NCHW"); tensor_desc_attr->set_layout("NCHW");
tensor_desc_attr->set_dtype(proto::DataType::DT_INT8);
tensor_desc_attr->set_dtype(ge::proto::DataType::DT_INT8);
auto attrs1 = tensor_desc_attr->mutable_attr(); auto attrs1 = tensor_desc_attr->mutable_attr();
auto attr1 = (*attrs1)["key2"].mutable_list(); // empty list attr auto attr1 = (*attrs1)["key2"].mutable_list(); // empty list attr


@@ -1219,14 +1219,14 @@ TEST(UTEST_ge_model_unserialize, test_invalid_TensorDesc) {
} }
TEST(UTEST_ge_model_unserialize, test_invalid_attr) { TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
{ // invalid graph { // invalid graph
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
auto graph_attr = attr_def->mutable_g(); auto graph_attr = attr_def->mutable_g();
auto attrs_of_graph = graph_attr->mutable_attr(); auto attrs_of_graph = graph_attr->mutable_attr();
auto tensor_val = (*attrs_of_graph)["key2"].mutable_td(); auto tensor_val = (*attrs_of_graph)["key2"].mutable_td();
tensor_val->set_dtype(proto::DT_INT8);
tensor_val->set_dtype(ge::proto::DT_INT8);
tensor_val->set_layout("invalidLayout"); tensor_val->set_layout("invalidLayout");


ModelSerializeImp imp; ModelSerializeImp imp;
@@ -1245,15 +1245,15 @@ TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8); EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8);
} }
{ // invalid list graph { // invalid list graph
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
attr_def->mutable_list()->set_val_type(ge::proto::AttrDef_ListValue_ListValueType_VT_LIST_GRAPH); attr_def->mutable_list()->set_val_type(ge::proto::AttrDef_ListValue_ListValueType_VT_LIST_GRAPH);
auto graph_attr = attr_def->mutable_list()->add_g(); auto graph_attr = attr_def->mutable_list()->add_g();
auto attrs_of_graph = graph_attr->mutable_attr(); auto attrs_of_graph = graph_attr->mutable_attr();
auto tensor_val = (*attrs_of_graph)["key2"].mutable_td(); auto tensor_val = (*attrs_of_graph)["key2"].mutable_td();
tensor_val->set_dtype(proto::DT_INT8);
tensor_val->set_dtype(ge::proto::DT_INT8);
tensor_val->set_layout("invalidLayout"); tensor_val->set_layout("invalidLayout");


ModelSerializeImp imp; ModelSerializeImp imp;
@@ -1273,14 +1273,14 @@ TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8); EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8);
} }
{ // invalid named_attrs { // invalid named_attrs
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
auto graph_attr = attr_def->mutable_func(); auto graph_attr = attr_def->mutable_func();
auto attrs_of_graph = graph_attr->mutable_attr(); auto attrs_of_graph = graph_attr->mutable_attr();
auto tensor_val = (*attrs_of_graph)["key2"].mutable_td(); auto tensor_val = (*attrs_of_graph)["key2"].mutable_td();
tensor_val->set_dtype(proto::DT_INT8);
tensor_val->set_dtype(ge::proto::DT_INT8);
tensor_val->set_layout("invalidLayout"); tensor_val->set_layout("invalidLayout");


ModelSerializeImp imp; ModelSerializeImp imp;
@@ -1298,15 +1298,15 @@ TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8); EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8);
} }
{ // invalid list named_attrs { // invalid list named_attrs
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
attr_def->mutable_list()->set_val_type(ge::proto::AttrDef_ListValue_ListValueType_VT_LIST_NAMED_ATTRS); attr_def->mutable_list()->set_val_type(ge::proto::AttrDef_ListValue_ListValueType_VT_LIST_NAMED_ATTRS);
auto graph_attr = attr_def->mutable_list()->add_na(); auto graph_attr = attr_def->mutable_list()->add_na();
auto attrs_of_graph = graph_attr->mutable_attr(); auto attrs_of_graph = graph_attr->mutable_attr();
auto tensor_val = (*attrs_of_graph)["key2"].mutable_td(); auto tensor_val = (*attrs_of_graph)["key2"].mutable_td();
tensor_val->set_dtype(proto::DT_INT8);
tensor_val->set_dtype(ge::proto::DT_INT8);
tensor_val->set_layout("invalidLayout"); tensor_val->set_layout("invalidLayout");


ModelSerializeImp imp; ModelSerializeImp imp;
@@ -1325,14 +1325,14 @@ TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8); EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8);
} }
{ // invalid tensor_desc { // invalid tensor_desc
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
auto graph_attr = attr_def->mutable_td(); auto graph_attr = attr_def->mutable_td();
auto attrs_of_graph = graph_attr->mutable_attr(); auto attrs_of_graph = graph_attr->mutable_attr();
auto tensor_val = (*attrs_of_graph)["key2"].mutable_td(); auto tensor_val = (*attrs_of_graph)["key2"].mutable_td();
tensor_val->set_dtype(proto::DT_INT8);
tensor_val->set_dtype(ge::proto::DT_INT8);
tensor_val->set_layout("invalidLayout"); tensor_val->set_layout("invalidLayout");


ModelSerializeImp imp; ModelSerializeImp imp;
@@ -1350,15 +1350,15 @@ TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8); EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8);
} }
{ // invalid list tensor_desc { // invalid list tensor_desc
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
attr_def->mutable_list()->set_val_type(ge::proto::AttrDef_ListValue_ListValueType_VT_LIST_TENSOR_DESC); attr_def->mutable_list()->set_val_type(ge::proto::AttrDef_ListValue_ListValueType_VT_LIST_TENSOR_DESC);
auto graph_attr = attr_def->mutable_list()->add_td(); auto graph_attr = attr_def->mutable_list()->add_td();
auto attrs_of_graph = graph_attr->mutable_attr(); auto attrs_of_graph = graph_attr->mutable_attr();
auto tensor_val = (*attrs_of_graph)["key2"].mutable_td(); auto tensor_val = (*attrs_of_graph)["key2"].mutable_td();
tensor_val->set_dtype(proto::DT_INT8);
tensor_val->set_dtype(ge::proto::DT_INT8);
tensor_val->set_layout("invalidLayout"); tensor_val->set_layout("invalidLayout");


ModelSerializeImp imp; ModelSerializeImp imp;
@@ -1377,14 +1377,14 @@ TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8); EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8);
} }
{ // invalid tensor { // invalid tensor
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
auto graph_attr = attr_def->mutable_t()->mutable_desc(); auto graph_attr = attr_def->mutable_t()->mutable_desc();
auto attrs_of_graph = graph_attr->mutable_attr(); auto attrs_of_graph = graph_attr->mutable_attr();
auto tensor_val = (*attrs_of_graph)["key2"].mutable_td(); auto tensor_val = (*attrs_of_graph)["key2"].mutable_td();
tensor_val->set_dtype(proto::DT_INT8);
tensor_val->set_dtype(ge::proto::DT_INT8);
tensor_val->set_layout("invalidLayout"); tensor_val->set_layout("invalidLayout");


ModelSerializeImp imp; ModelSerializeImp imp;
@@ -1402,15 +1402,15 @@ TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8); EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8);
} }
{ // invalid list tensor { // invalid list tensor
proto::ModelDef mode_def;
ge::proto::ModelDef mode_def;
auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr auto attrs = mode_def.add_graph()->add_op()->mutable_attr(); // node attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
attr_def->mutable_list()->set_val_type(ge::proto::AttrDef_ListValue_ListValueType_VT_LIST_TENSOR); attr_def->mutable_list()->set_val_type(ge::proto::AttrDef_ListValue_ListValueType_VT_LIST_TENSOR);
auto graph_attr = attr_def->mutable_list()->add_t()->mutable_desc(); auto graph_attr = attr_def->mutable_list()->add_t()->mutable_desc();
auto attrs_of_graph = graph_attr->mutable_attr(); auto attrs_of_graph = graph_attr->mutable_attr();
auto tensor_val = (*attrs_of_graph)["key2"].mutable_td(); auto tensor_val = (*attrs_of_graph)["key2"].mutable_td();
tensor_val->set_dtype(proto::DT_INT8);
tensor_val->set_dtype(ge::proto::DT_INT8);
tensor_val->set_layout("invalidLayout"); tensor_val->set_layout("invalidLayout");


ModelSerializeImp imp; ModelSerializeImp imp;
@@ -1429,15 +1429,15 @@ TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8); EXPECT_EQ(tensor_desc1.GetDataType(), DT_INT8);
} }
{ // invalid list tensor { // invalid list tensor
proto::GraphDef graph_def;
ge::proto::GraphDef graph_def;
auto attrs = graph_def.add_op()->mutable_attr(); // node attr auto attrs = graph_def.add_op()->mutable_attr(); // node attr


proto::AttrDef *attr_def = &(*attrs)["key1"];
ge::proto::AttrDef *attr_def = &(*attrs)["key1"];
attr_def->mutable_list()->set_val_type(ge::proto::AttrDef_ListValue_ListValueType_VT_LIST_TENSOR); attr_def->mutable_list()->set_val_type(ge::proto::AttrDef_ListValue_ListValueType_VT_LIST_TENSOR);
auto graph_attr = attr_def->mutable_list()->add_t()->mutable_desc(); auto graph_attr = attr_def->mutable_list()->add_t()->mutable_desc();
auto attrs_of_graph = graph_attr->mutable_attr(); auto attrs_of_graph = graph_attr->mutable_attr();
auto tensor_val = (*attrs_of_graph)["key2"].mutable_td(); auto tensor_val = (*attrs_of_graph)["key2"].mutable_td();
tensor_val->set_dtype(proto::DT_INT8);
tensor_val->set_dtype(ge::proto::DT_INT8);
tensor_val->set_layout("invalidLayout"); tensor_val->set_layout("invalidLayout");


ModelSerializeImp imp; ModelSerializeImp imp;
@@ -1462,7 +1462,7 @@ TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
TEST(UTEST_ge_model_unserialize, test_invalid_input_output) { TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
// model invalid node input // model invalid node input
{ {
proto::ModelDef model_def;
ge::proto::ModelDef model_def;
auto op_def = model_def.add_graph()->add_op(); // node attr auto op_def = model_def.add_graph()->add_op(); // node attr
op_def->add_input("invalidNodeName:0"); op_def->add_input("invalidNodeName:0");


@@ -1475,7 +1475,7 @@ TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
} }
// model invalid node control input // model invalid node control input
{ {
proto::ModelDef model_def;
ge::proto::ModelDef model_def;
auto op_def = model_def.add_graph()->add_op(); // node attr auto op_def = model_def.add_graph()->add_op(); // node attr
op_def->add_input("invalidNodeName:-1"); op_def->add_input("invalidNodeName:-1");


@@ -1488,7 +1488,7 @@ TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
} }
// model invalid graph input // model invalid graph input
{ {
proto::ModelDef model_def;
ge::proto::ModelDef model_def;
model_def.add_graph()->add_input("invalidNodeName:0"); model_def.add_graph()->add_input("invalidNodeName:0");


Buffer buffer(model_def.ByteSizeLong()); Buffer buffer(model_def.ByteSizeLong());
@@ -1500,7 +1500,7 @@ TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
} }
// model invalid graph input // model invalid graph input
{ {
proto::ModelDef model_def;
ge::proto::ModelDef model_def;
model_def.add_graph()->add_output("invalidNodeName:0"); model_def.add_graph()->add_output("invalidNodeName:0");


Buffer buffer(model_def.ByteSizeLong()); Buffer buffer(model_def.ByteSizeLong());
@@ -1512,7 +1512,7 @@ TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
} }
// graph invalid node input // graph invalid node input
{ {
proto::GraphDef graph_def;
ge::proto::GraphDef graph_def;
auto op_def = graph_def.add_op(); // node attr auto op_def = graph_def.add_op(); // node attr
op_def->add_input("invalidNodeName:0"); op_def->add_input("invalidNodeName:0");


@@ -1525,7 +1525,7 @@ TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
} }
// graph invalid node control input // graph invalid node control input
{ {
proto::GraphDef graph_def;
ge::proto::GraphDef graph_def;
auto op_def = graph_def.add_op(); // node attr auto op_def = graph_def.add_op(); // node attr
op_def->add_input("invalidNodeName:-1"); op_def->add_input("invalidNodeName:-1");


@@ -1538,7 +1538,7 @@ TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
} }
// graph invalid graph input // graph invalid graph input
{ {
proto::GraphDef graph_def;
ge::proto::GraphDef graph_def;
graph_def.add_input("invalidNodeName:0"); graph_def.add_input("invalidNodeName:0");


Buffer buffer(graph_def.ByteSizeLong()); Buffer buffer(graph_def.ByteSizeLong());
@@ -1550,7 +1550,7 @@ TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
} }
// graph invalid graph output // graph invalid graph output
{ {
proto::GraphDef graph_def;
ge::proto::GraphDef graph_def;
graph_def.add_output("invalidNodeName:0"); graph_def.add_output("invalidNodeName:0");


Buffer buffer(graph_def.ByteSizeLong()); Buffer buffer(graph_def.ByteSizeLong());
@@ -1562,7 +1562,7 @@ TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
} }
// model invalid node input anchor // model invalid node input anchor
{ {
proto::ModelDef model_def;
ge::proto::ModelDef model_def;
auto graph_def = model_def.add_graph(); auto graph_def = model_def.add_graph();
auto node_def1 = graph_def->add_op(); // node attr auto node_def1 = graph_def->add_op(); // node attr
node_def1->set_name("node1"); node_def1->set_name("node1");


+ 2
- 1
tests/ut/common/graph/testcase/ge_graph/ge_node_unittest.cc View File

@@ -151,7 +151,7 @@ TEST_F(UtestGeNode, update_opdesc) {


EXPECT_EQ(n1->UpdateOpDesc(desc_ptr2), GRAPH_SUCCESS); EXPECT_EQ(n1->UpdateOpDesc(desc_ptr2), GRAPH_SUCCESS);
} }
/*
TEST_F(UtestGeNode, add_link_from) { TEST_F(UtestGeNode, add_link_from) {
OpDescPtr desc_ptr = std::make_shared<OpDesc>("name", "type"); OpDescPtr desc_ptr = std::make_shared<OpDesc>("name", "type");
EXPECT_EQ(desc_ptr->AddInputDesc("x", GeTensorDesc(GeShape({1, 16, 16, 16}), FORMAT_NCHW)), GRAPH_SUCCESS); EXPECT_EQ(desc_ptr->AddInputDesc("x", GeTensorDesc(GeShape({1, 16, 16, 16}), FORMAT_NCHW)), GRAPH_SUCCESS);
@@ -179,6 +179,7 @@ TEST_F(UtestGeNode, add_link_from) {
NodePtr n8 = graph_ptr1->AddNode(desc_ptr1); NodePtr n8 = graph_ptr1->AddNode(desc_ptr1);
EXPECT_EQ(n8->AddLinkFromForParse(n7), GRAPH_PARAM_INVALID); EXPECT_EQ(n8->AddLinkFromForParse(n7), GRAPH_PARAM_INVALID);
} }
*/


TEST_F(UtestGeNode, add_link_from_fail) { TEST_F(UtestGeNode, add_link_from_fail) {
OpDescPtr desc_ptr = std::make_shared<OpDesc>("name1", "type1"); OpDescPtr desc_ptr = std::make_shared<OpDesc>("name1", "type1");


+ 319
- 320
tests/ut/ge/CMakeLists.txt View File

@@ -17,348 +17,347 @@ project(ut_ge)


set(CMAKE_CXX_STANDARD 11) set(CMAKE_CXX_STANDARD 11)


file(GLOB_RECURSE PROTO_LIST RELATIVE ${CMAKE_CURRENT_LIST_DIR}
"${GE_SOURCE_DIR}/src/proto/om.proto"
"${GE_SOURCE_DIR}/src/proto/ge_ir.proto"
"${GE_SOURCE_DIR}/src/proto/ge_api.proto"
"${GE_SOURCE_DIR}/src/proto/fwk_adapter.proto"
"${GE_SOURCE_DIR}/src/proto/op_mapping_info.proto"
"${GE_SOURCE_DIR}/src/proto/ge_api.proto"
"${onnx_INC}/onnx/onnx.proto"
set(PROTO_LIST
"${GE_CODE_DIR}/metadef/proto/om.proto"
"${GE_CODE_DIR}/metadef/proto/ge_ir.proto"
"${GE_CODE_DIR}/metadef/proto/ge_api.proto"
"${GE_CODE_DIR}/metadef/proto/fwk_adapter.proto"
"${GE_CODE_DIR}/metadef/proto/op_mapping_info.proto"
"${GE_CODE_DIR}/metadef/proto/ge_api.proto"
"${GE_CODE_DIR}/metadef/proto/proto_inner/ge_onnx.proto"
) )


ge_protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})
protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})


# include directories # include directories
include_directories(${CMAKE_CURRENT_LIST_DIR}) include_directories(${CMAKE_CURRENT_LIST_DIR})
include_directories(${GE_SOURCE_DIR})
include_directories(${GE_SOURCE_DIR}/src/ge/inc)
include_directories(${GE_SOURCE_DIR}/src)
include_directories(${GE_SOURCE_DIR}/src/ge)
include_directories(${GE_SOURCE_DIR}/src/common)
include_directories(${GE_SOURCE_DIR}/src/common/graph)
include_directories(${GE_SOURCE_DIR}/inc)
include_directories(${GE_SOURCE_DIR}/inc/external)
include_directories(${GE_SOURCE_DIR}/inc/external/graph)
include_directories(${GE_SOURCE_DIR}/inc/graph)
include_directories(${GE_SOURCE_DIR}/inc/framework)
include_directories(${GE_SOURCE_DIR}/inc/common)
include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc)
include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc/cce)
include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc/ops)
include_directories(${GE_SOURCE_DIR}/tests/ut/ge)
include_directories(${GE_CODE_DIR})
include_directories(${GE_CODE_DIR}/inc)
include_directories(${GE_CODE_DIR}/ge)
include_directories(${GE_CODE_DIR}/metadef)
include_directories(${GE_CODE_DIR}/metadef/graph)
include_directories(${GE_CODE_DIR}/inc/external)
include_directories(${GE_CODE_DIR}/metadef/inc/external/graph)
include_directories(${GE_CODE_DIR}/metadef/inc/graph)
include_directories(${GE_CODE_DIR}/inc/framework)
include_directories(${GE_CODE_DIR}/metadef/inc/common)
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc)
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/cce)
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/ops)
include_directories(${GE_CODE_DIR}/tests/ut/ge)
include_directories(${CMAKE_BINARY_DIR}) include_directories(${CMAKE_BINARY_DIR})
include_directories(${CMAKE_BINARY_DIR}/proto/ge) include_directories(${CMAKE_BINARY_DIR}/proto/ge)

file(GLOB_RECURSE COMMON_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/common/properties_manager.cc"
"${GE_SOURCE_DIR}/src/ge/common/ge/plugin_manager.cc"
"${GE_SOURCE_DIR}/src/ge/common/ge/tbe_plugin_manager.cc"
"${GE_SOURCE_DIR}/src/common/graph/option/ge_local_context.cc"
"${GE_SOURCE_DIR}/src/common/graph/option/ge_context.cc"
"${GE_SOURCE_DIR}/src/ge/common/types.cc"
"${GE_SOURCE_DIR}/src/ge/common/op_map.cc"
"${GE_SOURCE_DIR}/src/ge/common/fmk_error_codes.cc"
"${GE_SOURCE_DIR}/src/ge/common/op/ge_op_utils.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/util/node_searcher/need_rebuild_node_searcher.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/util/variable_accelerate_ctrl.cc"
"${GE_SOURCE_DIR}/src/ge/opskernel_manager/ops_kernel_manager.cc"
"${GE_SOURCE_DIR}/src/ge/generator/ge_generator.cc"
"${GE_SOURCE_DIR}/src/ge/generator/generator_api.cc"
"${GE_SOURCE_DIR}/src/ge/graph/common/omg_util.cc"
"${GE_SOURCE_DIR}/src/ge/graph/common/bcast.cc"
"${GE_SOURCE_DIR}/src/ge/common/util.cc"
"${GE_SOURCE_DIR}/src/common/graph/ge_attr_define.cc"
"${GE_SOURCE_DIR}/src/common/graph/anchor.cc"
"${GE_SOURCE_DIR}/src/common/graph/ge_attr_value.cc"
"${GE_SOURCE_DIR}/src/common/graph/attr_value.cc"
"${GE_SOURCE_DIR}/src/common/graph/buffer.cc"
"${GE_SOURCE_DIR}/src/common/graph/compute_graph.cc"
"${GE_SOURCE_DIR}/src/common/graph/graph.cc"
"${GE_SOURCE_DIR}/src/common/graph/inference_context.cc"
"${GE_SOURCE_DIR}/src/common/graph/shape_refiner.cc"
"${GE_SOURCE_DIR}/src/common/graph/model.cc"
"${GE_SOURCE_DIR}/src/common/graph/model_serialize.cc"
"${GE_SOURCE_DIR}/src/common/graph/node.cc"
"${GE_SOURCE_DIR}/src/common/graph/op_desc.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator_factory.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator_factory_impl.cc"
"${GE_SOURCE_DIR}/src/common/graph/operator_reg.cc"
"${GE_SOURCE_DIR}/src/common/graph/range_vistor.cc"
"${GE_SOURCE_DIR}/src/common/graph/ge_tensor.cc"
"${GE_SOURCE_DIR}/src/common/graph/tensor.cc"
"${GE_SOURCE_DIR}/src/common/graph/detail/attributes_holder.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/anchor_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/graph_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/ge_ir_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/node_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/op_desc_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/tensor_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/utils/type_utils.cc"
"${GE_SOURCE_DIR}/src/common/graph/debug/graph_debug.cc"
"${GE_SOURCE_DIR}/src/common/graph/opsproto/opsproto_manager.cc"
"${GE_SOURCE_DIR}/src/common/graph/op_imp.cc"
"${GE_SOURCE_DIR}/src/common/register/register.cc"
"${GE_SOURCE_DIR}/src/common/register/op_kernel_registry.cc"
"${GE_SOURCE_DIR}/tests/depends/cce/src/op_kernel_registry.cc"
"${GE_SOURCE_DIR}/src/common/register/auto_mapping_util.cc"
"${GE_SOURCE_DIR}/src/common/register/tensor_assign.cc"
"${GE_SOURCE_DIR}/src/common/graph/format_refiner.cc"
include_directories(${CMAKE_BINARY_DIR}/proto/ge/proto)

set(COMMON_SRC_FILES
"${GE_CODE_DIR}/ge/common/properties_manager.cc"
"${GE_CODE_DIR}/ge/common/ge/plugin_manager.cc"
"${GE_CODE_DIR}/ge/common/ge/tbe_plugin_manager.cc"
"${GE_CODE_DIR}/metadef/graph/option/ge_local_context.cc"
"${GE_CODE_DIR}/metadef/graph/option/ge_context.cc"
"${GE_CODE_DIR}/ge/common/types.cc"
"${GE_CODE_DIR}/parser/parser/common/op_map.cc"
"${GE_CODE_DIR}/ge/common/fmk_error_codes.cc"
"${GE_CODE_DIR}/ge/common/op/ge_op_utils.cc"
#"${GE_CODE_DIR}/ge/graph/manager/util/node_searcher/need_rebuild_node_searcher.cc"
"${GE_CODE_DIR}/ge/graph/manager/util/variable_accelerate_ctrl.cc"
"${GE_CODE_DIR}/ge/opskernel_manager/ops_kernel_manager.cc"
"${GE_CODE_DIR}/ge/generator/ge_generator.cc"
"${GE_CODE_DIR}/ge/generator/generator_api.cc"
"${GE_CODE_DIR}/ge/graph/common/omg_util.cc"
"${GE_CODE_DIR}/ge/graph/common/bcast.cc"
"${GE_CODE_DIR}/ge/common/util.cc"
"${GE_CODE_DIR}/metadef/graph/ge_attr_define.cc"
"${GE_CODE_DIR}/metadef/graph/anchor.cc"
"${GE_CODE_DIR}/metadef/graph/ge_attr_value.cc"
"${GE_CODE_DIR}/metadef/graph/attr_value.cc"
"${GE_CODE_DIR}/metadef/graph/buffer.cc"
"${GE_CODE_DIR}/metadef/graph/compute_graph.cc"
"${GE_CODE_DIR}/metadef/graph/graph.cc"
"${GE_CODE_DIR}/metadef/graph/inference_context.cc"
"${GE_CODE_DIR}/metadef/graph/shape_refiner.cc"
"${GE_CODE_DIR}/metadef/graph/model.cc"
"${GE_CODE_DIR}/metadef/graph/model_serialize.cc"
"${GE_CODE_DIR}/metadef/graph/node.cc"
"${GE_CODE_DIR}/metadef/graph/op_desc.cc"
"${GE_CODE_DIR}/metadef/graph/operator.cc"
"${GE_CODE_DIR}/metadef/graph/operator_factory.cc"
"${GE_CODE_DIR}/metadef/graph/operator_factory_impl.cc"
#"${GE_CODE_DIR}/metadef/graph/operator_reg.cc"
#"${GE_CODE_DIR}/metadef/graph/range_vistor.cc"
"${GE_CODE_DIR}/metadef/graph/ge_tensor.cc"
"${GE_CODE_DIR}/metadef/graph/tensor.cc"
"${GE_CODE_DIR}/metadef/graph/detail/attributes_holder.cc"
"${GE_CODE_DIR}/metadef/graph/utils/anchor_utils.cc"
"${GE_CODE_DIR}/metadef/graph/utils/graph_utils.cc"
"${GE_CODE_DIR}/metadef/graph/utils/ge_ir_utils.cc"
"${GE_CODE_DIR}/metadef/graph/utils/node_utils.cc"
"${GE_CODE_DIR}/metadef/graph/utils/op_desc_utils.cc"
"${GE_CODE_DIR}/metadef/graph/utils/tensor_utils.cc"
"${GE_CODE_DIR}/metadef/graph/utils/type_utils.cc"
"${GE_CODE_DIR}/metadef/graph/debug/graph_debug.cc"
"${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc"
"${GE_CODE_DIR}/metadef/ops/op_imp.cpp"
"${GE_CODE_DIR}/metadef/register/register.cpp"
"${GE_CODE_DIR}/metadef/register/op_kernel_registry.cpp"
#"${GE_CODE_DIR}/tests/depends/cce/src/op_kernel_registry.cpp"
"${GE_CODE_DIR}/metadef/register/auto_mapping_util.cpp"
"${GE_CODE_DIR}/metadef/register/tensor_assign.cpp"
"${GE_CODE_DIR}/metadef/graph/format_refiner.cc"
) )


file(GLOB_RECURSE COMMON_FORMAT_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/common/fp16_t.cc"
"${GE_SOURCE_DIR}/src/ge/common/ge_format_util.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/formats.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/datatype_transfer.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_transpose.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_nchw_nc1hwc0.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_fractal_z.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_fractal_nz.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_fractal_zz.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_nhwc_nc1hwc0.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_nc1hwc0_nchw.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_nc1hwc0_nhwc.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_hwcn_c1hwncoc0.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_c1hwncoc0_hwcn.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_fracz_nchw.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_fracz_nhwc.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/format_transfers/format_transfer_fracz_hwcn.cc"
"${GE_SOURCE_DIR}/src/ge/common/formats/utils/formats_trans_utils.cc"
set(COMMON_FORMAT_SRC_FILES
"${GE_CODE_DIR}/ge/common/fp16_t.cc"
"${GE_CODE_DIR}/ge/common/ge_format_util.cc"
"${GE_CODE_DIR}/ge/common/formats/formats.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/datatype_transfer.cc"
#"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_transpose.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_nchw_nc1hwc0.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fractal_z.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fractal_nz.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fractal_zz.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_nhwc_nc1hwc0.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_nc1hwc0_nchw.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_nc1hwc0_nhwc.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_hwcn_c1hwncoc0.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_c1hwncoc0_hwcn.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fracz_nchw.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fracz_nhwc.cc"
"${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fracz_hwcn.cc"
"${GE_CODE_DIR}/ge/common/formats/utils/formats_trans_utils.cc"
) )


file(GLOB_RECURSE GRAPH_OPTIMIZE_COMMON_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/graph/optimize/graph_optimize.cc"
"${GE_SOURCE_DIR}/src/ge/graph/optimize/summary_optimize.cc"
set(GRAPH_OPTIMIZE_COMMON_SRC_FILES
"${GE_CODE_DIR}/ge/graph/optimize/graph_optimize.cc"
"${GE_CODE_DIR}/ge/graph/optimize/summary_optimize.cc"
) )




file(GLOB_RECURSE GRAPH_PREPARE_COMMON_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/graph/preprocess/graph_preprocess.cc"
"${GE_SOURCE_DIR}/src/ge/graph/preprocess/insert_op/util_insert_aipp_op.cc"
"${GE_SOURCE_DIR}/src/ge/graph/preprocess/insert_op/ge_aipp_op.cc"
"${GE_SOURCE_DIR}/src/ge/graph/preprocess/insert_op/base_insert_op.cc"
set(GRAPH_PREPARE_COMMON_SRC_FILES
"${GE_CODE_DIR}/ge/graph/preprocess/graph_preprocess.cc"
"${GE_CODE_DIR}/ge/graph/preprocess/insert_op/util_insert_aipp_op.cc"
"${GE_CODE_DIR}/ge/graph/preprocess/insert_op/ge_aipp_op.cc"
#"${GE_CODE_DIR}/ge/graph/preprocess/insert_op/base_insert_op.cc"
) )


file(GLOB_RECURSE GRAPH_PARTITION_COMMON_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/graph/partition/graph_partition.cc"
"${GE_SOURCE_DIR}/src/ge/plugin/engine/dnnengines.cc"
"${GE_SOURCE_DIR}/src/ge/graph/partition/engine_place.cc"
set(GRAPH_PARTITION_COMMON_SRC_FILES
"${GE_CODE_DIR}/ge/graph/partition/graph_partition.cc"
"${GE_CODE_DIR}/ge/plugin/engine/dnnengines.cc"
"${GE_CODE_DIR}/ge/graph/partition/engine_place.cc"
) )


file(GLOB_RECURSE GRAPH_LOAD_COMMON_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/graph/load/graph_loader.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/graph_manager_utils.cc"
"${GE_SOURCE_DIR}/src/ge/omm/csa_interact.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/graph_mem_allocator.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/graph_var_manager.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/trans_var_data_utils.cc"
"${GE_SOURCE_DIR}/src/ge/common/util.cc"
set(GRAPH_LOAD_COMMON_SRC_FILES
"${GE_CODE_DIR}/ge/graph/load/graph_loader.cc"
"${GE_CODE_DIR}/ge/graph/manager/graph_manager_utils.cc"
"${GE_CODE_DIR}/ge/omm/csa_interact.cc"
"${GE_CODE_DIR}/ge/graph/manager/graph_mem_allocator.cc"
"${GE_CODE_DIR}/ge/graph/manager/graph_var_manager.cc"
"${GE_CODE_DIR}/ge/graph/manager/trans_var_data_utils.cc"
"${GE_CODE_DIR}/ge/common/util.cc"
) )


file(GLOB_RECURSE DISTINCT_GRAPH_LOAD_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/graph/manager/util/hcom_util.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/util/debug.cc"
"${GE_SOURCE_DIR}/src/ge/common/properties_manager.cc"
"${GE_SOURCE_DIR}/src/ge/common/profiling/profiling_manager.cc"
"${GE_SOURCE_DIR}/src/ge/common/model_parser/base.cc"
"${GE_SOURCE_DIR}/src/ge/common/tbe_kernel_store.cc"
"${GE_SOURCE_DIR}/src/ge/common/util.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/cpu_queue_schedule.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/data_dumper.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/data_inputer.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/davinci_model.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/davinci_model_parser.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/model_manager.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/model_output.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/model_utils.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/tbe_handle_store.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/event_record_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/event_wait_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/fusion_start_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/fusion_stop_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/hccl_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/kernel_ex_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/kernel_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/label_goto_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/label_set_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/memcpy_async_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/profiler_trace_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/stream_active_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/stream_switch_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/end_graph_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/model_exit_task_info.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/super_kernel/super_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/new_model_manager/task_info/super_kernel/super_kernel_factory.cc"
"${GE_SOURCE_DIR}/src/ge/graph/load/output/output.cc"
"${GE_SOURCE_DIR}/src/ge/model/ge_model.cc"
"${GE_SOURCE_DIR}/src/ge/common/helper/model_helper.cc"
"${GE_SOURCE_DIR}/src/ge/common/helper/om_file_helper.cc"
"${GE_SOURCE_DIR}/src/ge/common/debug/memory_dumper.cc"
"${GE_SOURCE_DIR}/src/ge/executor/ge_executor.cc"
"${GE_SOURCE_DIR}/src/ge/common/auth/file_saver.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/model_manager/event_manager.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/custom/custom_op.cc"
)
set(DISTINCT_GRAPH_LOAD_SRC_FILES
"${GE_CODE_DIR}/ge/graph/manager/util/hcom_util.cc"
"${GE_CODE_DIR}/ge/graph/manager/util/debug.cc"
"${GE_CODE_DIR}/ge/common/properties_manager.cc"
"${GE_CODE_DIR}/ge/common/profiling/profiling_manager.cc"
"${GE_CODE_DIR}/ge/common/model_parser/base.cc"
"${GE_CODE_DIR}/ge/common/tbe_kernel_store.cc"
"${GE_CODE_DIR}/ge/common/util.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/cpu_queue_schedule.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/data_dumper.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/data_inputer.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/davinci_model.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/davinci_model_parser.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/model_manager.cc"
#"${GE_CODE_DIR}/ge/graph/load/new_model_manager/model_output.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/model_utils.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/tbe_handle_store.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/event_record_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/event_wait_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/fusion_start_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/fusion_stop_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/hccl_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/kernel_ex_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/kernel_task_info.cc"
#"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/label_goto_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/label_set_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/memcpy_async_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/profiler_trace_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/stream_active_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/stream_switch_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/end_graph_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/model_exit_task_info.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/super_kernel/super_kernel.cc"
"${GE_CODE_DIR}/ge/graph/load/new_model_manager/task_info/super_kernel/super_kernel_factory.cc"
#"${GE_CODE_DIR}/ge/graph/load/output/output.cc"
"${GE_CODE_DIR}/ge/model/ge_model.cc"
"${GE_CODE_DIR}/ge/common/helper/model_helper.cc"
"${GE_CODE_DIR}/ge/common/helper/om_file_helper.cc"
"${GE_CODE_DIR}/ge/common/debug/memory_dumper.cc"
"${GE_CODE_DIR}/ge/executor/ge_executor.cc"
"${GE_CODE_DIR}/ge/common/auth/file_saver.cc"
"${GE_CODE_DIR}/ge/graph/manager/model_manager/event_manager.cc"
#"${GE_CODE_DIR}/ge/graph/manager/custom/custom_op.cc"
)


file(GLOB_RECURSE GRAPH_EXECUTE_COMMON_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/graph/execute/graph_execute.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/graph_manager.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/graph_context.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/util/rt_context_util.cc"
"${GE_SOURCE_DIR}/src/ge/graph/manager/graph_context.h"
"${GE_SOURCE_DIR}/src/ge/common/thread_pool.cc"
set(GRAPH_EXECUTE_COMMON_SRC_FILES
"${GE_CODE_DIR}/ge/graph/execute/graph_execute.cc"
"${GE_CODE_DIR}/ge/graph/manager/graph_manager.cc"
"${GE_CODE_DIR}/ge/graph/manager/graph_context.cc"
"${GE_CODE_DIR}/ge/graph/manager/util/rt_context_util.cc"
"${GE_CODE_DIR}/ge/graph/manager/graph_context.h"
"${GE_CODE_DIR}/ge/common/thread_pool.cc"
) )


file(GLOB_RECURSE GRAPH_BUILD_COMMON_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/graph/build/graph_build.cc"
"${GE_SOURCE_DIR}/src/ge/graph/build/new_model/task_generator.cc"
"${GE_SOURCE_DIR}/src/ge/init/gelib.cc"
"${GE_SOURCE_DIR}/src/ge/client/ge_api.cc"
"${GE_SOURCE_DIR}/src/ge/session/inner_session.cc"
"${GE_SOURCE_DIR}/src/ge/session/session_manager.cc"
"${GE_SOURCE_DIR}/src/ge/engine_manager/dnnengine_manager.cc"
"${GE_SOURCE_DIR}/src/ge/plugin/engine/engine_manage.cc"
"${GE_SOURCE_DIR}/src/ge/graph/build/logical_stream_allocator.cc"
"${GE_SOURCE_DIR}/src/ge/graph/build/stream_allocator.cc"
"${GE_SOURCE_DIR}/src/ge/graph/build/memory/block_mem_assigner.cc"
"${GE_SOURCE_DIR}/src/ge/graph/build/memory/binary_block_mem_assigner.cc"
"${GE_SOURCE_DIR}/src/ge/graph/build/memory/hybrid_mem_assigner.cc"
"${GE_SOURCE_DIR}/src/ge/graph/build/memory/max_block_mem_assigner.cc"
"${GE_SOURCE_DIR}/src/ge/model/ge_model.cc"
"${GE_SOURCE_DIR}/src/ge/common/helper/model_helper.cc"
"${GE_SOURCE_DIR}/src/ge/common/helper/om_file_helper.cc"
"${GE_SOURCE_DIR}/src/ge/common/tbe_kernel_store.cc"
"${GE_SOURCE_DIR}/src/ge/common/model_parser/base.cc"
"${GE_SOURCE_DIR}/src/ge/graph/build/run_context.cc"
"${GE_SOURCE_DIR}/src/ge/graph/build/optimize_stream_graph.cc"
set(GRAPH_BUILD_COMMON_SRC_FILES
"${GE_CODE_DIR}/ge/graph/build/graph_builder.cc"
"${GE_CODE_DIR}/ge/graph/build/task_generator.cc"
"${GE_CODE_DIR}/ge/init/gelib.cc"
"${GE_CODE_DIR}/ge/client/ge_api.cc"
"${GE_CODE_DIR}/ge/session/inner_session.cc"
"${GE_CODE_DIR}/ge/session/session_manager.cc"
"${GE_CODE_DIR}/ge/engine_manager/dnnengine_manager.cc"
"${GE_CODE_DIR}/ge/plugin/engine/engine_manage.cc"
"${GE_CODE_DIR}/ge/graph/build/logical_stream_allocator.cc"
"${GE_CODE_DIR}/ge/graph/build/stream_allocator.cc"
"${GE_CODE_DIR}/ge/graph/build/memory/block_mem_assigner.cc"
"${GE_CODE_DIR}/ge/graph/build/memory/binary_block_mem_assigner.cc"
"${GE_CODE_DIR}/ge/graph/build/memory/hybrid_mem_assigner.cc"
"${GE_CODE_DIR}/ge/graph/build/memory/max_block_mem_assigner.cc"
"${GE_CODE_DIR}/ge/model/ge_model.cc"
"${GE_CODE_DIR}/ge/common/helper/model_helper.cc"
"${GE_CODE_DIR}/ge/common/helper/om_file_helper.cc"
"${GE_CODE_DIR}/ge/common/tbe_kernel_store.cc"
"${GE_CODE_DIR}/ge/common/model_parser/base.cc"
"${GE_CODE_DIR}/ge/graph/build/run_context.cc"
#"${GE_CODE_DIR}/ge/graph/build/optimize_stream_graph.cc"
) )


file(GLOB_RECURSE GRAPH_PASS_COMMON_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/graph/passes/pass_manager.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/base_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/variable_prepare_op_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/variable_ref_delete_op_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/atomic_addr_clean_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/constant_folding_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/iterator_fusion_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/iterator_op_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/net_output_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/update_net_output_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/end_graph_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/node_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/print_op_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/dimension_compute_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/dimension_adjust_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/get_original_format_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/stop_gradient_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/guarantee_const_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/prevent_gradient_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/identity_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/placeholder_with_default_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/snapshot_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/shape_operate_op_remove_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/unused_op_remove_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/isolated_op_remove_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/permute_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/var_is_initialized_op_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/cast_translate_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/dropout_scope_split_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/prune_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/enter_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/next_iteration_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/switch_op_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/pass_utils.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/addn_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/save_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/merge_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/switch_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/switch_logic_remove_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/assert_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/dropout_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/unused_const_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/reshape_remove_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/resource_pair_add_control_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/resource_pair_remove_control_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/transop_breadth_fusion_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/transop_without_reshape_fusion_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/transop_depth_fusion_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/transop_nearby_allreduce_fusion_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/same_transdata_breadth_fusion_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/compile_nodes_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/common/transop_util.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/flow_ctrl_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/optimize/optimizer/allreduce_fusion_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/variable_op_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/transpose_transdata_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/hccl_memcpy_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/no_reshape_op_remove_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/no_use_reshape_remove_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/control_op_attr_pass.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/infershape_pass.cc"
"${GE_SOURCE_DIR}/src/ge/ge_local_engine/engine/host_cpu_engine.cc"
set(GRAPH_PASS_COMMON_SRC_FILES
"${GE_CODE_DIR}/ge/graph/passes/pass_manager.cc"
"${GE_CODE_DIR}/ge/graph/passes/base_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/variable_prepare_op_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/variable_ref_delete_op_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/atomic_addr_clean_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/constant_folding_pass.cc"
"${GE_CODE_DIR}/parser/parser/tensorflow/iterator_fusion_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/iterator_op_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/net_output_pass.cc"
#"${GE_CODE_DIR}/ge/graph/passes/update_net_output_pass.cc"
#"${GE_CODE_DIR}/ge/graph/passes/end_graph_pass.cc"
#"${GE_CODE_DIR}/ge/graph/passes/node_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/print_op_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/dimension_compute_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/dimension_adjust_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/get_original_format_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/stop_gradient_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/guarantee_const_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/prevent_gradient_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/identity_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/placeholder_with_default_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/snapshot_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/shape_operate_op_remove_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/unused_op_remove_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/isolated_op_remove_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/permute_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/var_is_initialized_op_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/cast_translate_pass.cc"
#"${GE_CODE_DIR}/ge/graph/passes/dropout_scope_split_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/prune_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/enter_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/next_iteration_pass.cc"
#"${GE_CODE_DIR}/ge/graph/passes/switch_op_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/pass_utils.cc"
"${GE_CODE_DIR}/ge/graph/passes/addn_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/save_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/merge_pass.cc"
#"${GE_CODE_DIR}/ge/graph/passes/switch_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/switch_logic_remove_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/assert_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/dropout_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/unused_const_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/reshape_remove_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/resource_pair_add_control_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/resource_pair_remove_control_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/transop_breadth_fusion_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/transop_without_reshape_fusion_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/transop_depth_fusion_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/transop_nearby_allreduce_fusion_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/same_transdata_breadth_fusion_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/compile_nodes_pass.cc"
"${GE_CODE_DIR}/ge/graph/common/transop_util.cc"
"${GE_CODE_DIR}/ge/graph/passes/flow_ctrl_pass.cc"
"${GE_CODE_DIR}/ge/graph/optimize/optimizer/allreduce_fusion_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/folding_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/variable_op_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/transpose_transdata_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/hccl_memcpy_pass.cc"
#"${GE_CODE_DIR}/ge/graph/passes/no_reshape_op_remove_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/no_use_reshape_remove_pass.cc"
#"${GE_CODE_DIR}/ge/graph/passes/control_op_attr_pass.cc"
"${GE_CODE_DIR}/ge/graph/passes/infershape_pass.cc"
"${GE_CODE_DIR}/ge/ge_local_engine/engine/host_cpu_engine.cc"
) )


file(GLOB_RECURSE KERNEL_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/broadcast_gradient_args_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/broadcast_args_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/greater_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/gather_v2_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/maximum_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/floormod_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/floordiv_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/range_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/mul_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/shape_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/shape_n_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/size_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/rank_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/fill_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/empty_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/cast_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/transdata_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/permute_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/squeeze_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/reshape_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/reformat_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/kernel_utils.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/expanddims_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/ssd_prior_box_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/pack_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/strided_slice_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/concat_v2_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/add_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/sub_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/reduce_prod_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/rsqrt_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/concat_offset_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/slice_kernel.cc"
"${GE_SOURCE_DIR}/src/ge/graph/passes/folding_kernel/dynamic_stitch_kernel.cc"
set(KERNEL_SRC_FILES
"${GE_CODE_DIR}/ge/host_kernels/broadcast_gradient_args_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/broadcast_args_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/greater_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/gather_v2_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/maximum_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/floormod_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/floordiv_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/range_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/mul_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/shape_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/shape_n_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/size_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/rank_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/fill_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/empty_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/cast_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/transdata_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/permute_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/squeeze_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/reshape_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/reformat_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/kernel_utils.cc"
"${GE_CODE_DIR}/ge/host_kernels/expanddims_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/ssd_prior_box_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/pack_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/strided_slice_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/concat_v2_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/add_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/sub_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/reduce_prod_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/rsqrt_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/concat_offset_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/slice_kernel.cc"
"${GE_CODE_DIR}/ge/host_kernels/dynamic_stitch_kernel.cc"
) )


file(GLOB_RECURSE SINGLE_OP_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"${GE_SOURCE_DIR}/src/ge/single_op/task/build_task_utils.cc"
"${GE_SOURCE_DIR}/src/ge/single_op/task/op_task.cc"
"${GE_SOURCE_DIR}/src/ge/single_op/task/tbe_task_builder.cc"
"${GE_SOURCE_DIR}/src/ge/single_op/single_op.cc"
"${GE_SOURCE_DIR}/src/ge/single_op/single_op_model.cc"
"${GE_SOURCE_DIR}/src/ge/single_op/stream_resource.cc"
"${GE_SOURCE_DIR}/src/ge/single_op/single_op_manager.cc"
set(SINGLE_OP_SRC_FILES
"${GE_CODE_DIR}/ge/single_op/task/build_task_utils.cc"
"${GE_CODE_DIR}/ge/single_op/task/op_task.cc"
"${GE_CODE_DIR}/ge/single_op/task/tbe_task_builder.cc"
"${GE_CODE_DIR}/ge/single_op/single_op.cc"
"${GE_CODE_DIR}/ge/single_op/single_op_model.cc"
"${GE_CODE_DIR}/ge/single_op/stream_resource.cc"
"${GE_CODE_DIR}/ge/single_op/single_op_manager.cc"
) )


# test files # test files
file(GLOB_RECURSE COMMON_TEST_FILES ${CMAKE_CURRENT_SOURCE_DIR}
set(COMMON_TEST_FILES
"graph/passes/graph_builder_utils.cc" "graph/passes/graph_builder_utils.cc"
"test.cc" "test.cc"
) )


file(GLOB_RECURSE DISTINCT_GRAPH_LOAD_TEST_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"graph/load/data_dumper_unittest.cc"
"graph/load/new_model_manager_data_inputer_unittest.cc"
set(DISTINCT_GRAPH_LOAD_TEST_FILES
"graph/load/data_dumper_unittest.cc"
"graph/load/new_model_manager_data_inputer_unittest.cc"
"graph/load/new_model_manager_davinci_model_unittest.cc" "graph/load/new_model_manager_davinci_model_unittest.cc"
"graph/load/new_model_manager_model_manager_unittest.cc" "graph/load/new_model_manager_model_manager_unittest.cc"
"graph/load/new_model_manager_task_build_unittest.cc" "graph/load/new_model_manager_task_build_unittest.cc"
@@ -370,7 +369,7 @@ file(GLOB_RECURSE DISTINCT_GRAPH_LOAD_TEST_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"graph/ge_executor_unittest.cc" "graph/ge_executor_unittest.cc"
) )


file(GLOB_RECURSE PASS_TEST_FILES ${CMAKE_CURRENT_SOURCE_DIR}
set(PASS_TEST_FILES
"graph/passes/prune_pass_unittest.cc" "graph/passes/prune_pass_unittest.cc"
"graph/passes/enter_pass_unittest.cc" "graph/passes/enter_pass_unittest.cc"
"graph/passes/switch_op_pass_unittest.cc" "graph/passes/switch_op_pass_unittest.cc"
@@ -413,7 +412,7 @@ file(GLOB_RECURSE PASS_TEST_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"graph/passes/infershape_pass_unittest.cc" "graph/passes/infershape_pass_unittest.cc"
) )


file(GLOB_RECURSE KERNEL_TEST_FILES ${CMAKE_CURRENT_SOURCE_DIR}
set(KERNEL_TEST_FILES
"graph/passes/folding_kernel/greater_kernel_unittest.cc" "graph/passes/folding_kernel/greater_kernel_unittest.cc"
"graph/passes/folding_kernel/maximum_kernel_unittest.cc" "graph/passes/folding_kernel/maximum_kernel_unittest.cc"
"graph/passes/folding_kernel/floormod_kernel_unittest.cc" "graph/passes/folding_kernel/floormod_kernel_unittest.cc"
@@ -449,7 +448,7 @@ file(GLOB_RECURSE KERNEL_TEST_FILES ${CMAKE_CURRENT_SOURCE_DIR}
"graph/passes/folding_kernel/dynamic_stitch_kernel_unittest.cc" "graph/passes/folding_kernel/dynamic_stitch_kernel_unittest.cc"
) )


file(GLOB_RECURSE MULTI_PARTS_TEST_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
set(MULTI_PARTS_TEST_FILES
"graph_ir/ge_operator_factory_unittest.cc" "graph_ir/ge_operator_factory_unittest.cc"
"graph/transop_util_unittest.cc" "graph/transop_util_unittest.cc"
"common/datatype_transfer_unittest.cc" "common/datatype_transfer_unittest.cc"
@@ -475,23 +474,23 @@ file(GLOB_RECURSE MULTI_PARTS_TEST_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
"graph/build/mem_assigner_unittest.cc" "graph/build/mem_assigner_unittest.cc"
) )


file(GLOB_RECURSE SINGLE_OP_TEST_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
set(SINGLE_OP_TEST_FILES
"single_op/single_op_model_unittest.cc" "single_op/single_op_model_unittest.cc"
"single_op/single_op_manager_unittest.cc" "single_op/single_op_manager_unittest.cc"
"single_op/stream_resource_unittest.cc" "single_op/stream_resource_unittest.cc"
) )


file(GLOB_RECURSE PROFILING_MNG_TEST_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
set(PROFILING_MNG_TEST_FILES
"profiling/ge_profiling_manager_unittest.cc" "profiling/ge_profiling_manager_unittest.cc"
) )


file(GLOB_RECURSE OTHERS_TEST_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
set(OTHERS_TEST_FILES
"plugin_manager/ge_util_unittest.cc" "plugin_manager/ge_util_unittest.cc"
) )


list(APPEND COMMON_SHARED_LIBRARIES list(APPEND COMMON_SHARED_LIBRARIES
omg_stub omg_stub
graphengine::securec
c_sec
slog_stub slog_stub
cce_ge_stub cce_ge_stub
runtime_stub runtime_stub
@@ -540,9 +539,9 @@ add_executable(ut_libge_multiparts_utest
target_link_libraries(ut_libge_multiparts_utest target_link_libraries(ut_libge_multiparts_utest
ge_build_common ge_load_common ge_build_common ge_execute_common ge_optimize_common ge_partition_common ge_pass_common ge_build_common ge_load_common ge_build_common ge_execute_common ge_optimize_common ge_partition_common ge_pass_common
ge_prepare_common ge_single_op ge_ut_common ge_prepare_common ge_single_op ge_ut_common
graphengine::gtest graphengine::gtest_main protobuf::protobuf rt dl
gtest gtest_main ascend_protobuf -lrt -ldl
) )
target_link_libraries(ut_libge_multiparts_utest ${COMMON_SHARED_LIBRARIES} protobuf::protobuf)
target_link_libraries(ut_libge_multiparts_utest ${COMMON_SHARED_LIBRARIES} ascend_protobuf)


# libge_others_utest # libge_others_utest
add_executable(ut_libge_others_utest add_executable(ut_libge_others_utest
@@ -554,9 +553,9 @@ add_executable(ut_libge_others_utest
) )
target_link_libraries(ut_libge_others_utest target_link_libraries(ut_libge_others_utest
ge_execute_common ge_load_common ge_pass_common ge_ut_common ge_execute_common ge_load_common ge_pass_common ge_ut_common
graphengine::gtest graphengine::gtest_main protobuf::protobuf rt dl
gtest gtest_main ascend_protobuf -lrt -ldl
) )
target_link_libraries(ut_libge_others_utest ${COMMON_SHARED_LIBRARIES} protobuf::protobuf)
target_link_libraries(ut_libge_others_utest ${COMMON_SHARED_LIBRARIES} ascend_protobuf)


# libge_kernel_utest # libge_kernel_utest
add_executable(ut_libge_kernel_utest add_executable(ut_libge_kernel_utest
@@ -567,9 +566,9 @@ add_executable(ut_libge_kernel_utest
) )
target_link_libraries(ut_libge_kernel_utest target_link_libraries(ut_libge_kernel_utest
ge_pass_common ge_ut_common ge_load_common ge_pass_common ge_ut_common ge_load_common
graphengine::gtest graphengine::gtest_main protobuf::protobuf rt dl
gtest gtest_main ascend_protobuf -lrt -ldl
) )
target_link_libraries(ut_libge_kernel_utest ${COMMON_SHARED_LIBRARIES} protobuf::protobuf)
target_link_libraries(ut_libge_kernel_utest ${COMMON_SHARED_LIBRARIES} ascend_protobuf)


# libge_distinct_load_utest # libge_distinct_load_utest
add_executable(ut_libge_distinct_load_utest add_executable(ut_libge_distinct_load_utest
@@ -583,5 +582,5 @@ target_link_libraries(ut_libge_distinct_load_utest ${COMMON_SHARED_LIBRARIES}
ge_execute_common ge_ut_common ge_ut_common_format ge_pass_common ge_load_common ge_execute_common ge_ut_common ge_ut_common_format ge_pass_common ge_load_common
ge_single_op ge_prepare_common ge_single_op ge_prepare_common
ge_optimize_common ge_build_common ge_partition_common ge_optimize_common ge_build_common ge_partition_common
graphengine::gtest graphengine::gtest_main protobuf::protobuf rt dl pthread
gtest gtest_main ascend_protobuf -lrt -ldl -lpthread
) )

Loading…
Cancel
Save