fix test_split_api

PaddlePaddle · Feb 17, 2022 · 07f275f · 07f275f · paddle-bot-old · Feb 17, 2022
1 parent 1f40d23
commit 07f275f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/paddle/pten/api/lib/manual_api.cc b/paddle/pten/api/lib/manual_api.cc
@@ -28,6 +28,7 @@ limitations under the License. */
 #include "paddle/pten/infermeta/unary.h"
 
 PT_DECLARE_KERNEL(copy, CPU, ALL_LAYOUT);
+PT_DECLARE_KERNEL(split, CPU, ALL_LAYOUT);
 
 #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
 PT_DECLARE_KERNEL(copy, GPU, ALL_LAYOUT);