2021-05-18 14:03:33 +08:00
|
|
|
diff --git a/tensorpipe/test/CMakeLists.txt b/tensorpipe/test/CMakeLists.txt
|
2023-04-19 11:26:40 +08:00
|
|
|
index 53e8643..09baaa0 100644
|
2021-05-18 14:03:33 +08:00
|
|
|
--- a/tensorpipe/test/CMakeLists.txt
|
|
|
|
+++ b/tensorpipe/test/CMakeLists.txt
|
2023-04-19 11:26:40 +08:00
|
|
|
@@ -93,25 +93,30 @@ if(TP_USE_CUDA)
|
2021-05-18 14:03:33 +08:00
|
|
|
channel/cuda_ipc/cuda_ipc_test.cc
|
|
|
|
)
|
|
|
|
endif()
|
|
|
|
-
|
2022-05-18 14:57:56 +08:00
|
|
|
- list(APPEND TP_TEST_SRCS
|
2021-05-18 14:03:33 +08:00
|
|
|
- channel/cuda_gdr/cuda_gdr_test.cc
|
|
|
|
- )
|
2022-05-18 14:57:56 +08:00
|
|
|
-
|
2021-05-18 14:03:33 +08:00
|
|
|
+ if(TP_ENABLE_CUDA_GDR)
|
2022-05-18 14:57:56 +08:00
|
|
|
+ list(APPEND TP_TEST_SRCS
|
2021-05-18 14:03:33 +08:00
|
|
|
+ channel/cuda_gdr/cuda_gdr_test.cc
|
|
|
|
+ )
|
|
|
|
+ endif()
|
2022-05-18 14:57:56 +08:00
|
|
|
cuda_add_library(tensorpipe_cuda_kernel channel/kernel.cu)
|
|
|
|
list(APPEND TP_TEST_LINK_LIBRARIES tensorpipe_cuda_kernel)
|
|
|
|
|
|
|
|
list(APPEND TP_TEST_LINK_LIBRARIES tensorpipe_cuda)
|
2021-05-18 14:03:33 +08:00
|
|
|
endif()
|
|
|
|
|
2022-05-18 14:57:56 +08:00
|
|
|
-add_subdirectory(${PROJECT_SOURCE_DIR}/third_party/googletest
|
|
|
|
- ${PROJECT_BINARY_DIR}/third_party/googletest EXCLUDE_FROM_ALL)
|
|
|
|
+find_package(GTest CONFIG REQUIRED)
|
2023-04-19 11:26:40 +08:00
|
|
|
+
|
|
|
|
+if (TARGET uv)
|
|
|
|
+ set(LIBUV_LIBRARY uv)
|
|
|
|
+else()
|
|
|
|
+ set(LIBUV_LIBRARY uv_a)
|
|
|
|
+endif()
|
2022-05-18 14:57:56 +08:00
|
|
|
|
|
|
|
list(APPEND TP_TEST_LINK_LIBRARIES
|
|
|
|
tensorpipe
|
|
|
|
- uv::uv
|
|
|
|
- gmock
|
|
|
|
- gtest_main)
|
2023-04-19 11:26:40 +08:00
|
|
|
+ ${LIBUV_LIBRARY}
|
2022-05-18 14:57:56 +08:00
|
|
|
+ GTest::gmock
|
|
|
|
+ GTest::gtest_main)
|
|
|
|
|
|
|
|
add_executable(tensorpipe_test ${TP_TEST_SRCS})
|
2021-05-18 14:03:33 +08:00
|
|
|
|