Skip to content

Commit eddf6d0

Browse files
disable strided split (#56882)
* disable strided split
1 parent 0e74bf3 commit eddf6d0

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

paddle/phi/kernels/stride/split_kernel.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ void SplitWithNumStridedKernel(const Context& dev_ctx,
6565

6666
} // namespace phi
6767
PD_REGISTER_KERNEL_FOR_ALL_BACKEND_DTYPE_EXCEPT_CUSTOM(
68-
split, STRIDED, phi::SplitStridedKernel) {}
68+
split_strided, STRIDED, phi::SplitStridedKernel) {}
6969

7070
PD_REGISTER_KERNEL_FOR_ALL_BACKEND_DTYPE_EXCEPT_CUSTOM(
71-
split_with_num, STRIDED, phi::SplitWithNumStridedKernel) {}
71+
split_with_num_strided, STRIDED, phi::SplitWithNumStridedKernel) {}

test/legacy_test/test_stride.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -609,11 +609,11 @@ def call_stride(self):
609609
self.call_flatten()
610610
self.call_squeeze()
611611
self.call_unsqueeze()
612-
self.call_split()
613-
self.call_split2()
614-
self.call_split3()
615-
self.call_split4()
616-
self.call_chunk()
612+
# self.call_split()
613+
# self.call_split2()
614+
# self.call_split3()
615+
# self.call_split4()
616+
# self.call_chunk()
617617
self.call_unbind()
618618
self.call_as_strided()
619619
self.call_view()

0 commit comments

Comments
 (0)