• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1message(STATUS "Compiling GraphEngine")
2message(STATUS "[ME] build_path: ${BUILD_PATH}")
3
4function(ge_protobuf_generate c_var h_var)
5    common_protobuf_generate(${CMAKE_BINARY_DIR}/proto/ge/proto ${c_var} ${h_var} ${ARGN})
6    set(${c_var} ${${c_var}} PARENT_SCOPE)
7    set(${h_var} ${${h_var}} PARENT_SCOPE)
8endfunction()
9
10if(MSLITE_ENABLE_ACL)
11    set(GRAPHENGINE_PATH "${TOP_DIR}/graphengine/910")
12else()
13    set(GRAPHENGINE_PATH "${CMAKE_SOURCE_DIR}/graphengine/910")
14endif()
15
16if(DEFINED ASCEND_VERSION)
17    if(${ASCEND_VERSION} STREQUAL "910" OR ${ASCEND_VERSION} STREQUAL "910b")
18        add_definitions(-DASCEND_910)
19    endif()
20endif()
21
22if(ENABLE_TESTCASES OR MODE_ASCEND_ALL OR MODE_ASCEND_ACL)
23    if(NOT(BUILD_LITE))
24        file(GLOB_RECURSE GE_PROTO_FILE ${GRAPHENGINE_PATH}/metadef/proto/*.proto)
25    else()
26        file(GLOB_RECURSE GE_PROTO_FILE ${TOP_DIR}/graphengine/910/metadef/proto/*.proto)
27    endif()
28    set(TMP_FILE_NAME_LIST)
29    foreach(file ${GE_PROTO_FILE})
30        get_filename_component(file_name ${file} NAME_WE)
31        list(FIND TMP_FILE_NAME_LIST ${file_name} OUT_VAR)
32        if(NOT ${OUT_VAR} EQUAL "-1")
33            list(REMOVE_ITEM GE_PROTO_FILE ${file})
34        endif()
35        list(APPEND TMP_FILE_NAME_LIST ${file_name})
36    endforeach()
37    ge_protobuf_generate(GE_PROTO_SRCS GE_PROTO_HDRS ${GE_PROTO_FILE})
38    add_library(graph SHARED ${GE_PROTO_SRCS})
39    if(NOT MSVC)
40        set_target_properties(graph PROPERTIES COMPILE_FLAGS "-Wno-unused-veriable -Wno-array-bounds")
41    endif()
42else()
43    message(FATAL_ERROR "No compile option defined for GraphEngine, exiting")
44endif()
45