forked from huawei/mindspore2022
52 lines
1.7 KiB
CMake
52 lines
1.7 KiB
CMake
file(GLOB_RECURSE _PREACTIVATE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
|
"common/*.cc"
|
|
"mem_reuse/*.cc"
|
|
"pass/*.cc"
|
|
"somas/*.cc"
|
|
)
|
|
|
|
if("${ENABLE_HIDDEN}" STREQUAL "OFF")
|
|
string(REPLACE " -Werror " " " CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
|
|
string(REPLACE " -fvisibility=hidden" " -fvisibility=default" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
|
|
endif()
|
|
|
|
if(ENABLE_D OR ENABLE_ACL)
|
|
file(GLOB_RECURSE _D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
|
"ascend/*.cc"
|
|
)
|
|
list(APPEND _PREACTIVATE_SRC_LIST ${_D_SRC_LIST})
|
|
endif()
|
|
|
|
if(ENABLE_GPU)
|
|
file(GLOB_RECURSE _GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
|
"gpu/*.cc"
|
|
)
|
|
list(APPEND _PREACTIVATE_SRC_LIST ${_GPU_SRC_LIST})
|
|
endif()
|
|
|
|
if(ENABLE_GPU_INFER)
|
|
file(GLOB_RECURSE GPU_SRC_TRT_PASS_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "trt_pass/*.cc")
|
|
list(APPEND _PREACTIVATE_SRC_LIST ${GPU_SRC_TRT_PASS_LIST})
|
|
endif()
|
|
|
|
if(${CMAKE_SYSTEM_NAME} MATCHES "Darwin")
|
|
set(CMAKE_CXX_FLAGS_RELEASE "$ENV{CXXFLAGS} -Wno-user-defined-warnings -Wno-inconsistent-missing-override \
|
|
-Wno-overloaded-virtual -Wno-unused-const-variable -Wno-pessimizing-move")
|
|
endif()
|
|
|
|
if(ENABLE_CPU)
|
|
file(GLOB_RECURSE _CPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "cpu/*.cc")
|
|
list(APPEND _PREACTIVATE_SRC_LIST ${_CPU_SRC_LIST})
|
|
endif()
|
|
|
|
if(ENABLE_AKG AND ${CMAKE_SYSTEM_NAME} MATCHES "Linux")
|
|
file(GLOB_RECURSE _GK_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
|
"graph_kernel/*.cc"
|
|
)
|
|
list(APPEND _PREACTIVATE_SRC_LIST ${_GK_SRC_LIST})
|
|
endif()
|
|
|
|
set_property(SOURCE ${_PREACTIVATE_SRC_LIST} PROPERTY COMPILE_DEFINITIONS
|
|
SUBMODULE_ID=mindspore::SubModuleId::SM_PRE_ACT)
|
|
add_library(_mindspore_backend_optimizer_obj OBJECT ${_PREACTIVATE_SRC_LIST})
|