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

disable strided split (#56882)

* disable strided split
上级 0e74bf36
...@@ -65,7 +65,7 @@ void SplitWithNumStridedKernel(const Context& dev_ctx, ...@@ -65,7 +65,7 @@ void SplitWithNumStridedKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PD_REGISTER_KERNEL_FOR_ALL_BACKEND_DTYPE_EXCEPT_CUSTOM( 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( 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): ...@@ -609,11 +609,11 @@ class TestStride(unittest.TestCase):
self.call_flatten() self.call_flatten()
self.call_squeeze() self.call_squeeze()
self.call_unsqueeze() self.call_unsqueeze()
self.call_split() # self.call_split()
self.call_split2() # self.call_split2()
self.call_split3() # self.call_split3()
self.call_split4() # self.call_split4()
self.call_chunk() # self.call_chunk()
self.call_unbind() self.call_unbind()
self.call_as_strided() self.call_as_strided()
self.call_view() self.call_view()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册