CMakeLists.txt

# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# See LICENSE for license information.

add_executable(test_operator
               test_qdq.cu
               test_cast_transpose.cu
               test_transpose.cu
               test_cast_transpose_dbias.cu
               test_cast_transpose_dbias_dgelu.cu
               test_cast_transpose_dgeglu.cu
               test_gelu.cu
               test_geglu.cu
               test_dgeglu.cu
               test_layernorm.cu
               test_rmsnorm.cu
               test_multi_cast_transpose.cu
               ../test_common.cu)

list(APPEND test_operator_LINKER_LIBS CUDA::cudart GTest::gtest_main ${TE_LIB})

if(EXISTS ${NVTE_MPI_INCLUDE})
    list(APPEND test_operator_LINKER_LIBS ${MPI_LIB})
endif()

target_link_libraries(test_operator PUBLIC ${test_operator_LINKER_LIBS})
target_compile_options(test_operator PRIVATE -O2)

include(GoogleTest)
gtest_discover_tests(test_operator)