CMakeLists.txt

cc_test(
  test_math_function
  SRCS test_math_function.cc
  DEPS math_function)
if(WITH_GPU)
  nv_test(
    test_math_function_gpu
    SRCS test_math_function.cu
    DEPS math_function)
  nv_test(
    test_broadcast_gpu
    SRCS test_ternary_broadcast.cu
    DEPS gtest)
endif()
if(WITH_ROCM)
  hip_test(
    test_math_function_gpu
    SRCS test_math_function.cu
    DEPS math_function)
endif()

cc_test(
  test_cpu_vec
  SRCS test_cpu_vec.cc
  DEPS blas phi_backends)

# For String Kernels
cc_test(
  test_strings_lower_upper_dev_api
  SRCS test_strings_lower_upper_dev_api.cc
  DEPS phi)
if(WITH_GPU)
  nv_test(
    test_strings_lower_upper_dev_gpu_api
    SRCS test_strings_lower_upper_dev_api.cu
    DEPS phi)
elseif(WITH_ROCM)
  hip_test(
    test_strings_lower_upper_dev_gpu_api
    SRCS test_strings_lower_upper_dev_api.cu
    DEPS phi)
endif()

cc_test(
  test_strings_copy_dev_api
  SRCS test_strings_copy_dev_api.cc
  DEPS phi)
if(WITH_GPU)
  nv_test(
    test_strings_copy_dev_gpu_api
    SRCS test_strings_copy_dev_api.cu
    DEPS phi)
elseif(WITH_ROCM)
  hip_test(
    test_strings_copy_dev_gpu_api
    SRCS test_strings_copy_dev_api.cu
    DEPS phi)
endif()

cc_test(
  test_memcpy_dev_api
  SRCS test_memcpy_dev_api.cc
  DEPS phi)

cc_test(
  test_transfer_layout_dev_api
  SRCS test_transfer_layout_dev_api.cc
  DEPS phi)

if(WITH_GPU)
  nv_test(
    test_gpu_timer
    SRCS test_gpu_timer.cu
    DEPS gtest)
  nv_test(
    test_auto_tune
    SRCS test_auto_tune.cu
    DEPS gtest)
  cc_test(
    test_fused_adam_kernel
    SRCS test_fused_adam_kernel.cc
    DEPS gtest phi)
elseif(WITH_ROCM)
  hip_test(
    test_gpu_timer
    SRCS test_gpu_timer.cu
    DEPS gtest)
  hip_test(
    test_auto_tune
    SRCS test_auto_tune.cu
    DEPS gtest)
endif()

cc_test(
  test_cache
  SRCS test_cache.cc
  DEPS gtest cache)

cc_test(
  strided_memcpy_test
  SRCS strided_memcpy_test.cc
  DEPS phi_backends memory)

cc_test(
  sequence_padding_test
  SRCS sequence_padding_test.cc
  DEPS sequence_padding)

cc_test(
  sequence_pooling_test
  SRCS sequence_pooling_test.cc
  DEPS sequence_pooling)