Browse Source

fix(mgb/gopt): add ShuffleShuffleRemovePass assert

GitOrigin-RevId: 69e07c3197
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
9f4060b050
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      src/gopt/impl/tensor_reformat.cpp

+ 3
- 0
src/gopt/impl/tensor_reformat.cpp View File

@@ -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);
}


Loading…
Cancel
Save