From 9f4060b05008c63cbc83904e4d51803d206b644a Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 31 Jul 2020 10:57:03 +0800
Subject: [PATCH] fix(mgb/gopt): add ShuffleShuffleRemovePass assert

GitOrigin-RevId: 69e07c3197c4b44bfdfedae5e6630745daee166e
---
 src/gopt/impl/tensor_reformat.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/gopt/impl/tensor_reformat.cpp b/src/gopt/impl/tensor_reformat.cpp
index b6eefc157..ec8ad015c 100644
--- a/src/gopt/impl/tensor_reformat.cpp
+++ b/src/gopt/impl/tensor_reformat.cpp
@@ -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);
             }
-- 
GitLab