Update CMakeLists.txt

PaddlePaddle · Oct 11, 2023 · 1680119 · 1680119
1 parent 48ca42e
commit 1680119
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/test/cpp/fluid/fused/CMakeLists.txt b/test/cpp/fluid/fused/CMakeLists.txt
@@ -1,7 +1,6 @@
 if(WITH_CINN)
   set(CINN_DEPS cinn_compiler python)
 endif()
-get_property(paddle_lib GLOBAL PROPERTY PADDLE_LIB_NAME)
 if(WITH_GPU OR WITH_ROCM)
   # fusion_group
   if(NOT APPLE AND NOT WIN32)
@@ -34,7 +33,15 @@ if(WITH_GPU OR WITH_ROCM)
     nv_test(
       test_fused_layernorm_residual_dropout_bias
       SRCS fused_layernorm_residual_dropout_bias_test.cu
-      DEPS ${paddle_lib})
+      DEPS fleet_executor
+           tensor
+           op_registry
+           dropout_op
+           generated_op
+           device_context
+           phi
+           memory
+           ${CINN_DEPS})
   endif()
   # resnet_unit needs cudnn 8.0 above
   if((NOT WITH_ROCM) AND (NOT ${CUDNN_VERSION} VERSION_LESS 8000))