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: 7df661fbaf
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '7df661fbaf'
${ noResults }
mindspore2022/mindspore/ccsrc/runtime/hardware/CMakeLists.txt

18 lines
744 B
Raw Blame History 

  1. file(GLOB_RECURSE HARDWARE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  2.     "device_context_manager.cc")

  3. 

  4. if(ENABLE_GPU)

  5.     file(GLOB_RECURSE HARDWARE_GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  6.         "gpu/gpu_device_context.cc")

  7. endif()

  8. 

  9. if(ENABLE_CPU)

  10.     file(GLOB_RECURSE HARDWARE_CPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  11.         "cpu/cpu_device_context.cc")

  12. endif()

  13. 

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

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

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

  17.             ${HARDWARE_GPU_SRC_LIST} ${HARDWARE_CPU_SRC_LIST})