CMakeLists.txt 583 B

123456789101112131415161718192021222324252627
  1. cmake_minimum_required(VERSION 3.5)
  2. project(NVMLpp LANGUAGES CXX)
  3. set(CMAKE_CXX_STANDARD 17)
  4. set(CMAKE_CXX_STANDARD_REQUIRED ON)
  5. if("${CMAKE_CXX_COMPILER}" MATCHES "mingw")
  6. set(BUILD_WITH_MINGW ON)
  7. endif()
  8. file(GLOB_RECURSE HEADERS include/*.hpp)
  9. file(GLOB_RECURSE SOURCES src/*.cpp)
  10. include_directories(include/)
  11. include_directories(lib/)
  12. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
  13. add_library(${PROJECT_NAME} STATIC ${SOURCES} ${HEADERS})
  14. if(BUILD_WITH_MINGW)
  15. target_link_libraries(${PROJECT_NAME} nvml)
  16. else()
  17. target_link_libraries(${PROJECT_NAME} nvidia-ml)
  18. endif()