disable strided split (#56882)

* disable strided split
PaddlePaddle · Sep 4, 2023 · eddf6d0 · eddf6d0
1 parent 0e74bf3
commit eddf6d0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/paddle/phi/kernels/stride/split_kernel.cc b/paddle/phi/kernels/stride/split_kernel.cc
@@ -65,7 +65,7 @@ void SplitWithNumStridedKernel(const Context& dev_ctx,
 
 }  // namespace phi
 PD_REGISTER_KERNEL_FOR_ALL_BACKEND_DTYPE_EXCEPT_CUSTOM(
-    split, STRIDED, phi::SplitStridedKernel) {}
+    split_strided, STRIDED, phi::SplitStridedKernel) {}
 
 PD_REGISTER_KERNEL_FOR_ALL_BACKEND_DTYPE_EXCEPT_CUSTOM(
-    split_with_num, STRIDED, phi::SplitWithNumStridedKernel) {}
+    split_with_num_strided, STRIDED, phi::SplitWithNumStridedKernel) {}
diff --git a/test/legacy_test/test_stride.py b/test/legacy_test/test_stride.py
@@ -609,11 +609,11 @@ def call_stride(self):
         self.call_flatten()
         self.call_squeeze()
         self.call_unsqueeze()
-        self.call_split()
-        self.call_split2()
-        self.call_split3()
-        self.call_split4()
-        self.call_chunk()
+        # self.call_split()
+        # self.call_split2()
+        # self.call_split3()
+        # self.call_split4()
+        # self.call_chunk()
         self.call_unbind()
         self.call_as_strided()
         self.call_view()