Fixed CI failure with test_egr_task_eager_utils,test=document_fix (#4…

…0854)
PaddlePaddle · Mar 23, 2022 · 81e505d · 81e505d
1 parent b1a4668
commit 81e505d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/paddle/fluid/eager/tests/task_tests/eager_utils_test.cc b/paddle/fluid/eager/tests/task_tests/eager_utils_test.cc
@@ -24,6 +24,9 @@
 #include "paddle/fluid/eager/utils.h"
 
 #include "paddle/phi/api/lib/utils/allocator.h"
+#include "paddle/phi/core/kernel_registry.h"
+
+PD_DECLARE_KERNEL(full, CPU, ALL_LAYOUT);
 
 namespace egr {