!193 Mindspore support to compile at ascendcl

Merge pull request !193 from zhoufeng/origin_r1.0.1
pull/193/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 42d217fb8c

@ -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_DRIVER_DIR ${ASCEND_DIR}/driver/lib64/common ${ASCEND_DIR}/driver/lib64)
set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/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})

@ -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}) if (NOT ENABLE_ACL)
target_compile_definitions(ge_runner PRIVATE 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