30 lines
844 B
CMake
30 lines
844 B
CMake
# -----------------------------------------------------------------------------
|
|
# cuBLAS unittests
|
|
# -----------------------------------------------------------------------------
|
|
list(APPEND TF_CUBLAS_UNITTESTS
|
|
helper
|
|
level1
|
|
level2
|
|
level3
|
|
)
|
|
|
|
foreach(test IN LISTS TF_CUBLAS_UNITTESTS)
|
|
add_executable(${test} ${test}.cu)
|
|
target_link_libraries(${test}
|
|
${PROJECT_NAME} Threads::Threads tf::default_settings ${CUBLAS_LIBRARIES}
|
|
)
|
|
target_include_directories(${test} PRIVATE ${CUBLAS_INCLUDE_DIRS})
|
|
target_include_directories(${test} PRIVATE ${TF_3RD_PARTY_DIR}/doctest)
|
|
|
|
# avoid cmake 3.18+ warning
|
|
# we let nvcc to decide the flag if the architecture is not given
|
|
if(NOT CUDA_ARCHITECTURES)
|
|
set_property(TARGET ${test} PROPERTY CUDA_ARCHITECTURES OFF)
|
|
endif()
|
|
|
|
doctest_discover_tests(${test})
|
|
endforeach()
|
|
|
|
|
|
|
|
|