From eddf6d05857f5b549b5ee4018af13f037dc8e94b Mon Sep 17 00:00:00 2001 From: wanghuancoder Date: Mon, 4 Sep 2023 16:19:01 +0800 Subject: [PATCH] disable strided split (#56882) * disable strided split --- paddle/phi/kernels/stride/split_kernel.cc | 4 ++-- test/legacy_test/test_stride.py | 10 +++++----- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/paddle/phi/kernels/stride/split_kernel.cc b/paddle/phi/kernels/stride/split_kernel.cc index b264e21c72b..b468f77fbd8 100644 --- 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 index 2886e46febd..531575a2872 100644 --- a/test/legacy_test/test_stride.py +++ b/test/legacy_test/test_stride.py @@ -609,11 +609,11 @@ class TestStride(unittest.TestCase): 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() -- GitLab