zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: a64920c468
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'a64920c468'
${ noResults }
mindspore2022/mindspore/ccsrc/runtime/hardware/CMakeLists.txt

50 lines
2.5 kB
Raw Blame History 

  1. file(GLOB_RECURSE HARDWARE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  2.     "device_context_manager.cc" "collective/*.cc")

  3. 

  4. if(ENABLE_D)

  5.     file(GLOB_RECURSE HARDWARE_D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "ascend/*.cc")

  6. endif()

  7. 

  8. if(ENABLE_GPU)

  9.     file(GLOB_RECURSE HARDWARE_GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "gpu/*.cc")

  10.     list(REMOVE_ITEM HARDWARE_GPU_SRC_LIST "gpu/nvidia_collective_comm_lib.cc" "gpu/nvidia_communication_group.cc")

  11.     if(ENABLE_MPI)

  12.         set(NVIDIA_COLLECTIVE_SRCS "gpu/nvidia_collective_comm_lib.cc"

  13.                                 "gpu/nvidia_communication_group.cc"

  14.                                 "collective/collective_communication_lib.cc"

  15.                                 "collective/communication_group.cc")

  16.         set_property(SOURCE ${NVIDIA_COLLECTIVE_SRCS} PROPERTY COMPILE_DEFINITIONS

  17.                         SUBMODULE_ID=mindspore::SubModuleId::SM_DEVICE)

  18.         add_library(nvidia_collective SHARED ${NVIDIA_COLLECTIVE_SRCS})

  19.         target_link_libraries(nvidia_collective PRIVATE mindspore::nccl)

  20.     endif()

  21. endif()

  22. 

  23. if(ENABLE_CPU)

  24.     file(GLOB_RECURSE HARDWARE_CPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "cpu/*.cc")

  25.     list(REMOVE_ITEM HARDWARE_CPU_SRC_LIST "cpu/mpi_collective_comm_lib.cc" "cpu/mpi_communication_group.cc")

  26.     if(WIN32)

  27.         list(REMOVE_ITEM HARDWARE_CPU_SRC_LIST "cpu/ms_collective_comm_lib.cc")

  28.     endif()

  29.     if(ENABLE_MPI)

  30.         set(MPI_COLLECTIVE_SRCS "cpu/mpi_collective_comm_lib.cc"

  31.                                 "cpu/mpi_communication_group.cc"

  32.                                 "collective/collective_communication_lib.cc"

  33.                                 "collective/communication_group.cc")

  34.         set_property(SOURCE ${MPI_COLLECTIVE_SRCS} PROPERTY COMPILE_DEFINITIONS

  35.                         SUBMODULE_ID=mindspore::SubModuleId::SM_DEVICE)

  36.         add_library(mpi_collective SHARED ${MPI_COLLECTIVE_SRCS})

  37.         target_link_libraries(mpi_collective PRIVATE mindspore::ompi)

  38.     endif()

  39. endif()

  40. 

  41. if(${CMAKE_SYSTEM_NAME} MATCHES "Darwin")

  42.     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-delete-abstract-non-virtual-dtor")

  43. endif()

  44. 

  45. 

  46. set_property(SOURCE ${HARDWARE_SRC_LIST} ${HARDWARE_D_SRC_LIST} ${HARDWARE_GPU_SRC_LIST} ${HARDWARE_CPU_SRC_LIST}

  47.                     PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_DEVICE)

  48. add_library(_mindspore_runtime_hardware_obj OBJECT ${HARDWARE_SRC_LIST} ${HARDWARE_D_SRC_LIST}

  49.             ${HARDWARE_GPU_SRC_LIST} ${HARDWARE_CPU_SRC_LIST})