提交 9f4060b0 编写于 作者: M Megvii Engine Team

fix(mgb/gopt): add ShuffleShuffleRemovePass assert

GitOrigin-RevId: 69e07c3197c4b44bfdfedae5e6630745daee166e
上级 4495c0cc
......@@ -2924,6 +2924,9 @@ void ShuffleShuffleRemovePass::Impl::do_replace() {
#endif
auto new_var = rewriter.get_var(inp_var);
if (inp_format != out_format) {
mgb_assert(m_reformat.find(std::make_pair(
inp_format, out_format)) != m_reformat.end(),
"Unsupported shuffle shuffle remove pass");
new_var = m_reformat[std::make_pair(inp_format, out_format)](
new_var);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册