zhoufeng
5 years ago
parent
7afa6d0974
commit
b681cec8f2
File diff suppressed because it is too large
Load Diff
@ -1,2 +1,2 @@
|
||||
|
||||
add_library(_mindspore_common_obj OBJECT ${CMAKE_CURRENT_SOURCE_DIR}/*.cc)
|
||||
file(GLOB_RECURSE _COMMON_ALL_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")
|
||||
add_library(_mindspore_common_obj OBJECT ${_COMMON_ALL_SRC_FILES})
|
||||
|
||||
@ -1,12 +1,16 @@
|
||||
|
||||
set(_DEBUG_SRC_LIST
|
||||
"${CMAKE_CURRENT_SOURCE_DIR}/anf_ir_dump.cc"
|
||||
"${CMAKE_CURRENT_SOURCE_DIR}/anf_ir_utils.cc"
|
||||
"${CMAKE_CURRENT_SOURCE_DIR}/draw.cc"
|
||||
"${CMAKE_CURRENT_SOURCE_DIR}/dump_proto.cc")
|
||||
"${CMAKE_CURRENT_SOURCE_DIR}/dump_proto.cc"
|
||||
"${CMAKE_CURRENT_SOURCE_DIR}/info.cc"
|
||||
"${CMAKE_CURRENT_SOURCE_DIR}/label.cc"
|
||||
"${CMAKE_CURRENT_SOURCE_DIR}/trace_info.cc"
|
||||
"${CMAKE_CURRENT_SOURCE_DIR}/trace.cc"
|
||||
)
|
||||
|
||||
if(ENABLE_DUMP_E2E)
|
||||
if (ENABLE_DUMP_E2E)
|
||||
list(APPEND _DEBUG_SRC_LIST "${CMAKE_CURRENT_SOURCE_DIR}/e2e_dump.cc")
|
||||
endif(ENABLE_DUMP_E2E)
|
||||
endif (ENABLE_DUMP_E2E)
|
||||
|
||||
add_library(_mindspore_debug_obj OBJECT ${_DEBUG_SRC_LIST})
|
||||
@ -1,34 +1,50 @@
|
||||
file(GLOB_RECURSE _DEVICE_ALL_SRC_FILES *.cc)
|
||||
add_library(_mindspore_device_obj OBJECT ${_DEVICE_ALL_SRC_FILES})
|
||||
|
||||
if(ENABLE_CPU)
|
||||
target_compile_definitions(_mindspore_device_obj PRIVATE CPUSESSION)
|
||||
file(GLOB_RECURSE _CPU_SRC_LIST cpu/*.cc)
|
||||
add_library(_c_expression_cpu_device_obj OBJECT ${_CPU_SRC_LIST})
|
||||
endif()
|
||||
|
||||
if(ENABLE_GPU)
|
||||
file(GLOB_RECURSE _GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"gpu/*.cc"
|
||||
"gpu/*.cu"
|
||||
)
|
||||
list(REMOVE_ITEM _GPU_SRC_LIST "gpu/blocking_queue.cc"
|
||||
"gpu/gpu_buffer_mgr.cc"
|
||||
"gpu/mpi/mpi_initializer.cc"
|
||||
file(GLOB_RECURSE DEVICE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "common/*.cc"
|
||||
"kernel_info.cc" "kernel_runtime.cc" "memory_manager.cc" "kernel_runtime_manager.cc" "convert_tensor_utils.cc"
|
||||
)
|
||||
|
||||
if (ENABLE_GPU)
|
||||
list(APPEND DEVICE_SRC_LIST "gpu/distribution/collective_init.cc")
|
||||
else ()
|
||||
list(APPEND DEVICE_SRC_LIST "gpu/distribution/collective_fake_init.cc")
|
||||
endif ()
|
||||
|
||||
if (ENABLE_D)
|
||||
file(GLOB_RECURSE D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "ascend/*.cc" "kernel_adjust.cc")
|
||||
endif ()
|
||||
|
||||
if (ENABLE_CPU)
|
||||
file(GLOB_RECURSE CPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "cpu/*.cc")
|
||||
endif ()
|
||||
|
||||
# gpu
|
||||
if (ENABLE_GPU)
|
||||
file(GLOB_RECURSE CUDA_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "gpu/*.cc" "gpu/*.cu")
|
||||
|
||||
# gpu_queue
|
||||
list(REMOVE_ITEM CUDA_SRC_LIST "gpu/blocking_queue.cc" "gpu/gpu_buffer_mgr.cc")
|
||||
add_library(gpu_queue SHARED "gpu/blocking_queue.cc" "gpu/gpu_buffer_mgr.cc")
|
||||
target_link_libraries(gpu_queue ${CMAKE_THREAD_LIBS_INIT} ${CUDA_PATH}/lib64/libcudart.so)
|
||||
|
||||
list(REMOVE_ITEM CUDA_SRC_LIST "gpu/mpi/mpi_initializer.cc"
|
||||
"gpu/distribution/collective_wrapper.cc"
|
||||
"gpu/distribution/mpi_wrapper.cc"
|
||||
"gpu/distribution/nccl_wrapper.cc")
|
||||
add_library(_cuda_gpu_device_obj OBJECT ${_GPU_SRC_LIST})
|
||||
endif()
|
||||
|
||||
if(ENABLE_D)
|
||||
file(GLOB_RECURSE _D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"ascend/*.cc"
|
||||
"ascend/profiling/*.cc"
|
||||
"ascend/tasksink/*.cc"
|
||||
"kernel_adjust.cc"
|
||||
"ascend/tasksink/taskinfo/*.cc"
|
||||
"gpu/distribution/nccl_wrapper.cc"
|
||||
)
|
||||
target_sources(_mindspore_device_obj PRIVATE ${_D_SRC_LIST})
|
||||
endif()
|
||||
|
||||
if (ENABLE_MPI)
|
||||
include(ExternalProject)
|
||||
# gpu_collective
|
||||
add_library(gpu_collective SHARED "gpu/distribution/collective_wrapper.cc"
|
||||
"gpu/distribution/mpi_wrapper.cc"
|
||||
"gpu/distribution/nccl_wrapper.cc"
|
||||
)
|
||||
# _ms_mpi
|
||||
pybind11_add_module(_ms_mpi "gpu/mpi/mpi_initializer.cc")
|
||||
target_link_libraries(_ms_mpi PRIVATE mindspore::pybind11_module mindspore::ompi)
|
||||
target_link_libraries(gpu_collective PRIVATE mindspore::ompi mindspore::nccl)
|
||||
endif ()
|
||||
|
||||
# add_library(_mindspore_device_cuda_obj OBJECT ${CUDA_SRC_LIST})
|
||||
endif ()
|
||||
|
||||
add_library(_mindspore_device_obj OBJECT ${DEVICE_SRC_LIST} ${D_SRC_LIST} ${CPU_SRC_LIST})
|
||||
|
||||
@ -0,0 +1,5 @@
|
||||
file(GLOB_RECURSE MS_GVAR_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} *.cc)
|
||||
add_library(mindspore_gvar SHARED ${MS_GVAR_SRC_LIST})
|
||||
if (APPLE)
|
||||
set_target_properties(mindspore_gvar PROPERTIES MACOSX_RPATH ON)
|
||||
endif ()
|
||||
@ -1,5 +1,2 @@
|
||||
file(GLOB_RECURSE _IR_ALL_SRC_FILES
|
||||
./*.cc
|
||||
dtype/*.cc)
|
||||
|
||||
add_library(_mindspore_ir_obj OBJECT ${_IR_ALL_SRC_FILES})
|
||||
file(GLOB_RECURSE _IR_SRC_LIST ./*.cc dtype/*.cc)
|
||||
add_library(_mindspore_ir_obj OBJECT ${_IR_SRC_LIST})
|
||||
|
||||
@ -1,34 +1,40 @@
|
||||
file(GLOB_RECURSE _SESSION_ALL_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"kernel_query.cc"
|
||||
"kernel_fusion.cc"
|
||||
file(GLOB_RECURSE KERNEL_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"kernel_build_info.cc"
|
||||
"kash/*.cc"
|
||||
"common_utils.cc"
|
||||
"oplib/*.cc"
|
||||
)
|
||||
)
|
||||
|
||||
add_library(_mindspore_kernel_obj OBJECT ${_SESSION_ALL_SRC_FILES})
|
||||
if (ENABLE_D)
|
||||
file(GLOB_RECURSE D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"kernel_query.cc"
|
||||
"kernel_fusion.cc"
|
||||
"tbe/*.cc"
|
||||
"aicpu/*.cc"
|
||||
"mng/*.cc"
|
||||
"hccl/*.cc"
|
||||
)
|
||||
add_compile_definitions(ENABLE_D)
|
||||
endif ()
|
||||
|
||||
if(ENABLE_GPU)
|
||||
file(GLOB_RECURSE _CUDA_GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
if (ENABLE_GPU)
|
||||
file(GLOB_RECURSE CUDA_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"gpu/*.cu"
|
||||
"akg/gpu/*.cc"
|
||||
"akg/akgkernelbuild.cc"
|
||||
"akg/akg_kernel_attrs_process.cc"
|
||||
)
|
||||
add_library(_cuda_gpu_kernel_obj OBJECT ${_CUDA_GPU_SRC_LIST})
|
||||
|
||||
file(GLOB_RECURSE _C_EXPRESSION_GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"gpu/*.cc"
|
||||
)
|
||||
list(REMOVE_ITEM _C_EXPRESSION_GPU_SRC_LIST "gpu/nccl/nccl_gpu_kernel.cc")
|
||||
add_library(_c_expression_gpu_device_obj OBJECT ${_C_EXPRESSION_GPU_SRC_LIST})
|
||||
endif()
|
||||
file(GLOB_RECURSE GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "gpu/*.cc")
|
||||
list(REMOVE_ITEM GPU_SRC_LIST "gpu/nccl/nccl_gpu_kernel.cc")
|
||||
|
||||
if(ENABLE_D)
|
||||
file(GLOB_RECURSE _D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"tbe/*.cc"
|
||||
"aicpu/*.cc"
|
||||
"mng/*.cc"
|
||||
"hccl/*.cc"
|
||||
)
|
||||
target_sources(_mindspore_kernel_obj PRIVATE ${_D_SRC_LIST})
|
||||
if (ENABLE_MPI)
|
||||
include(ExternalProject)
|
||||
file(GLOB_RECURSE GPU_NCCL_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "gpu/nccl/*.cc")
|
||||
list(APPEND GPU_SRC_LIST ${GPU_NCCL_LIST})
|
||||
endif ()
|
||||
|
||||
# add_library(_mindspore_kernel_cuda_obj OBJECT ${CUDA_SRC_LIST})
|
||||
endif()
|
||||
|
||||
add_library(_mindspore_kernel_obj OBJECT ${KERNEL_SRC_LIST} ${GPU_SRC_LIST} ${D_SRC_LIST})
|
||||
|
||||
@ -1,5 +1,2 @@
|
||||
file(GLOB_RECURSE _ONNX_ALL_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"*.cc"
|
||||
)
|
||||
|
||||
add_library(_mindspore_onnx_obj OBJECT ${_ONNX_ALL_SRC_FILES})
|
||||
file(GLOB_RECURSE _ONNX_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")
|
||||
add_library(_mindspore_onnx_obj OBJECT ${_ONNX_SRC_FILES})
|
||||
|
||||
@ -1,5 +1,2 @@
|
||||
file(GLOB_RECURSE _OPERATOR_ALL_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"*.cc"
|
||||
)
|
||||
|
||||
add_library(_mindspore_operator_obj OBJECT ${_OPERATOR_ALL_SRC_FILES})
|
||||
file(GLOB_RECURSE _OPERATOR_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")
|
||||
add_library(_mindspore_operator_obj OBJECT ${_OPERATOR_SRC_FILES})
|
||||
|
||||
@ -1,9 +1,2 @@
|
||||
file(GLOB_RECURSE _OPTIMIZER_ALL_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"*.cc"
|
||||
)
|
||||
|
||||
add_library(_mindspore_optimizer_obj OBJECT ${_OPTIMIZER_ALL_SRC_FILES})
|
||||
if(ENABLE_DUMP_PROTO)
|
||||
file(GLOB_RECURSE _PROTO_SRC_LIST "parallel/strategy_checkpoint/parallel_strategy_checkpoint.cc")
|
||||
target_sources(_mindspore_optimizer_obj PRIVATE ${_PROTO_SRC_LIST})
|
||||
endif()
|
||||
file(GLOB_RECURSE _OPTIMIZER_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")
|
||||
add_library(_mindspore_optimizer_obj OBJECT ${_OPTIMIZER_SRC_FILES})
|
||||
|
||||
@ -0,0 +1,6 @@
|
||||
file(GLOB_RECURSE _PARALLEL_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")
|
||||
if (ENABLE_DUMP_PROTO)
|
||||
list(REMOVE_ITEM _PARALLEL_SRC_FILES "parallel/strategy_checkpoint/parallel_strategy_checkpoint.cc")
|
||||
endif ()
|
||||
|
||||
add_library(_mindspore_parallel_obj OBJECT ${_PARALLEL_SRC_FILES})
|
||||
@ -0,0 +1,13 @@
|
||||
file(GLOB_RECURSE _PREACTIVATE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"common/*.cc"
|
||||
"mem_reuse/*.cc"
|
||||
"pass/*.cc"
|
||||
"gpu/*.cc"
|
||||
)
|
||||
|
||||
if (ENABLE_D)
|
||||
file(GLOB_RECURSE _D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "ascend/*.cc")
|
||||
list(APPEND _PREACTIVATE_SRC_LIST ${_D_SRC_LIST})
|
||||
endif ()
|
||||
|
||||
add_library(_mindspore_pre_activate_obj OBJECT ${_PREACTIVATE_SRC_LIST})
|
||||
@ -1,8 +1,14 @@
|
||||
file(GLOB_RECURSE _PRE_ACTIVE_ALL_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"ascend/*.cc"
|
||||
"common/*.cc"
|
||||
"pass/*.cc"
|
||||
"gpu/*.cc"
|
||||
)
|
||||
file(GLOB_RECURSE _PREDICT_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"predict.cc"
|
||||
"generator/utils/ir_model_util.cc"
|
||||
"converter/*.cc"
|
||||
"converter/attr_utils/*.cc"
|
||||
"converter/lite_model/*.cc"
|
||||
"converter/lite_model/operations/*.cc"
|
||||
)
|
||||
|
||||
add_library(_mindspore_pre_active_obj OBJECT ${_PRE_ACTIVE_ALL_SRC_FILES})
|
||||
if (ENABLE_D)
|
||||
file(GLOB_RECURSE _D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "generator/ir/*.cc")
|
||||
list(APPEND _PREDICT_SRC_LIST ${_D_SRC_LIST})
|
||||
endif ()
|
||||
add_library(_mindspore_predict_obj OBJECT ${_PREDICT_SRC_LIST})
|
||||
@ -1,5 +1,2 @@
|
||||
file(GLOB_RECURSE _PYNATIVE_ALL_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
"*.cc"
|
||||
)
|
||||
|
||||
add_library(_mindspore_pynative_obj OBJECT ${_PYNATIVE_ALL_SRC_FILES})
|
||||
file(GLOB_RECURSE _PYBIND_API_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")
|
||||
add_library(_mindspore_pybind_api_obj OBJECT ${_PYBIND_API_SRC_LIST})
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue