Spaces:
Runtime error
Runtime error
File size: 1,326 Bytes
e202b16 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
include(cmake/Cuda.cmake)
# TODO(tgale): Move cuSPARSE, cuBLAS deps to test & benchmark only.
cuda_find_library(CUDART_LIBRARY cudart_static)
cuda_find_library(CUBLAS_LIBRARY cublas_static)
cuda_find_library(CUSPARSE_LIBRARY cusparse_static)
list(APPEND SPUTNIK_LIBS "cudart_static;cublas_static;cusparse_static;culibos;cublasLt_static")
# Google Glog.
find_package(Glog REQUIRED)
list(APPEND SPUTNIK_LIBS ${GLOG_LIBRARIES})
if (BUILD_TEST)
# Google Abseil.
add_subdirectory(third_party/abseil-cpp)
# Google Test and Google Mock.
add_subdirectory(third_party/googletest)
set(BUILD_GTEST ON CACHE INTERNAL "Build gtest submodule.")
set(BUILD_GMOCK ON CACHE INTERNAL "Build gmock submodule.")
include_directories(SYSTEM ${PROJECT_SOURCE_DIR}/third_party/googletest/googletest/include)
include_directories(SYSTEM ${PROJECT_SOURCE_DIR}/third_party/googletest/googlemock/include)
list(APPEND SPUTNIK_TEST_LIBS "gtest_main;gmock;absl::random_random")
endif()
if (BUILD_BENCHMARK)
# Google Benchmark.
add_subdirectory(third_party/benchmark)
set(BENCHMARK_ENABLE_TESTING OFF CACHE INTERNAL "Build benchmark test suite.")
include_directories(SYSTEM ${PROJECT_SOURCE_DIR}/third_party/benchmark/include)
list(APPEND SPUTNIK_BENCHMARK_LIBS "gtest;absl::random_random;benchmark::benchmark_main")
endif()
|