33 lines
830 B
Text
33 lines
830 B
Text
|
# -----------------------------------------------------------------------------
|
||
|
|
# cuda examples
|
||
|
|
# -----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
list(APPEND TF_CUDA_EXAMPLES
|
||
|
|
cuda_device_property
|
||
|
|
cuda_saxpy
|
||
|
|
cuda_matmul
|
||
|
|
cuda_knn
|
||
|
|
cuda_capturer
|
||
|
|
cuda_reduce
|
||
|
|
cuda_scan
|
||
|
|
cuda_merge
|
||
|
|
cuda_sort
|
||
|
|
cuda_transform
|
||
|
|
cuda_rebind
|
||
|
|
cuda_find
|
||
|
|
)
|
||
|
|
|
||
|
|
foreach(cuda_example IN LISTS TF_CUDA_EXAMPLES)
|
||
|
|
add_executable(${cuda_example} ${cuda_example}.cu)
|
||
|
|
target_link_libraries(${cuda_example}
|
||
|
|
${PROJECT_NAME} ${ATOMIC_LIBRARY} Threads::Threads tf::default_settings
|
||
|
|
)
|
||
|
|
|
||
|
|
# 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 ${cuda_example} PROPERTY CUDA_ARCHITECTURES OFF)
|
||
|
|
endif()
|
||
|
|
endforeach()
|
||
|
|
|