Browse Source

Mindspore support to compile at ascendcl

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
pull/193/head
zhoufeng 4 years ago
parent
commit
20929c8bd3
2 changed files with 7 additions and 5 deletions
  1. +2
    -2
      CMakeLists.txt
  2. +5
    -3
      src/ge/CMakeLists.txt

+ 2
- 2
CMakeLists.txt View File

@@ -81,8 +81,8 @@ else()
else() else()
set(ASCEND_DIR /usr/local/Ascend) set(ASCEND_DIR /usr/local/Ascend)
endif() endif()
set(ASCEND_DRIVER_DIR ${ASCEND_DIR}/driver/lib64/common)
set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64)
set(ASCEND_DRIVER_DIR ${ASCEND_DIR}/driver/lib64/common ${ASCEND_DIR}/driver/lib64)
set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64 ${ASCEND_DIR}/acllib/lib64 ${ASCEND_DIR}/atc/lib64)
find_library(slog libslog.so ${ASCEND_DRIVER_DIR}) find_library(slog libslog.so ${ASCEND_DRIVER_DIR})
find_library(mmpa libmmpa.so ${ASCEND_DRIVER_DIR}) find_library(mmpa libmmpa.so ${ASCEND_DRIVER_DIR})
find_library(msprof libmsprofiler.a ${ASCEND_RUNTIME_DIR}) find_library(msprof libmsprofiler.a ${ASCEND_RUNTIME_DIR})


+ 5
- 3
src/ge/CMakeLists.txt View File

@@ -212,13 +212,14 @@ file(GLOB TRAIN_SRC_LIST RELATIVE ${CMAKE_CURRENT_LIST_DIR}




######### libge_runner.so ############# ######### libge_runner.so #############
add_library(ge_runner SHARED ${TRAIN_SRC_LIST} ${PROTO_SRCS} ${PROTO_CLIENT_SRCS} ${PROTO_HEADER_HDRS})
target_compile_definitions(ge_runner PRIVATE
if (NOT ENABLE_ACL)
add_library(ge_runner SHARED ${TRAIN_SRC_LIST} ${PROTO_SRCS} ${PROTO_CLIENT_SRCS} ${PROTO_HEADER_HDRS})
target_compile_definitions(ge_runner PRIVATE
PROTOBUF_INLINE_NOT_IN_HEADERS=0 PROTOBUF_INLINE_NOT_IN_HEADERS=0
DAVINCI_SUPPORT_PROFILING DAVINCI_SUPPORT_PROFILING
REUSE_MEMORY=1 REUSE_MEMORY=1
DAVINCI_CLOUD) DAVINCI_CLOUD)
target_link_libraries(ge_runner
target_link_libraries(ge_runner
graph graph
ge_common ge_common
ge_memory ge_memory
@@ -236,6 +237,7 @@ target_link_libraries(ge_runner
${msprofiler} ${msprofiler}
rt rt
dl) dl)
endif ()


######### libge_compiler.so ############# ######### libge_compiler.so #############
# need to remove dependencies on pb files later # need to remove dependencies on pb files later


Loading…
Cancel
Save