OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
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: b5bf56e0ee
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'b5bf56e0ee'
${ noResults }
MegEngine/dnn/cuda-stub/CMakeLists.txt

26 lines
729 B
Raw Blame History 

  1. file (GLOB_RECURSE CUDA_STUB src/libcuda.cpp)

  2. file (GLOB_RECURSE NVRTC_STUB src/libnvrtc.cpp)

  3. 

  4. if(MGE_WITH_CUDA_STUB)

  5.     list(APPEND STUB_SRC ${CUDA_STUB})

  6. endif()

  7. 

  8. if(MGE_WITH_NVRTC_STUB)

  9.     list(APPEND STUB_SRC ${NVRTC_STUB})

  10. endif()

  11. 

  12. if(MSVC OR WIN32)

  13.     add_library (cuda-stub STATIC ${STUB_SRC})

  14. else()

  15.     add_library (cuda-stub SHARED ${STUB_SRC})

  16. endif()

  17. 

  18. set_target_properties(cuda-stub PROPERTIES OUTPUT_NAME cuda_stub)

  19. target_compile_definitions(cuda-stub PRIVATE __CUDA_API_VERSION_INTERNAL)

  20. if (MSVC OR WIN32)

  21.     target_link_libraries(cuda-stub PRIVATE -Wl,--no-undefined)

  22. else()

  23.     target_link_libraries(cuda-stub PRIVATE dl -Wl,--no-undefined)

  24. endif()

  25. install (TARGETS cuda-stub EXPORT ${MGE_EXPORT_TARGETS})