未验证 提交 eddf6d05 编写于 作者: W wanghuancoder 提交者: GitHub

disable strided split (#56882)

* disable strided split
上级 0e74bf36
......@@ -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) {}
......@@ -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()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册