wehub-resource-sync
21 lines
401 B
CMake
21 lines
401 B
CMake
if(WITH_GPU)
|
|
if(${CMAKE_CUDA_COMPILER_VERSION} LESS 11.0)
|
|
nv_test(
|
|
check_reduce_rank_test
|
|
SRCS check_reduce_rank_test.cu
|
|
DEPS tensor cub)
|
|
else()
|
|
nv_test(
|
|
check_reduce_rank_test
|
|
SRCS check_reduce_rank_test.cu
|
|
DEPS tensor)
|
|
endif()
|
|
endif()
|
|
|
|
if(WITH_ROCM)
|
|
hip_test(
|
|
check_reduce_rank_test
|
|
SRCS check_reduce_rank_test.cu
|
|
DEPS tensor)
|
|
endif()
|