add custom test

PaddlePaddle · Jul 18, 2022 · a52d4f5 · a52d4f5
1 parent c4f9abd
commit a52d4f5
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/paddle/phi/backends/CMakeLists.txt b/paddle/phi/backends/CMakeLists.txt
@@ -45,3 +45,14 @@ target_link_libraries(phi_backends ${BACKENDS_DEPS})
 get_property(phi_modules GLOBAL PROPERTY PHI_MODULES)
 set(phi_modules ${phi_modules} phi_backends)
 set_property(GLOBAL PROPERTY PHI_MODULES "${phi_modules}")
+
+if(WITH_CUSTOM_DEVICE)
+  cc_test(
+    custom_device_test
+    SRCS custom/custom_device_test.cc
+    DEPS phi_backends phi_device_context)
+  cc_test(
+    capi_test
+    SRCS custom/capi_test.cc
+    DEPS phi_capi)
+endif()