Fixed matcher (#4351)

Co-authored-by: Sang Ik Lee <sang.ik.lee@intel.com>

Fixed matcher (#4351)
Co-authored-by: Sang Ik Lee <sang.ik.lee@intel.com>
cbf84017 · Ilya Churaev · GitHub · e5258c0a · cbf84017
Unverified Commit cbf84017 authored Feb 21, 2020 by Ilya Churaev Committed by GitHub Feb 21, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 12 additions and 4 deletions

matcher.cpp src/ngraph/pattern/matcher.cpp +12 -4

No files found.
--- a/src/ngraph/pattern/matcher.cpp
+++ b/src/ngraph/pattern/matcher.cpp
@@ -40,10 +40,18 @@ namespace ngraph
        {
            if (m_restore)
            {
-                m_matcher->m_matched_list.erase(m_matcher->m_matched_list.begin() + m_watermark,
-                                                m_matcher->m_matched_list.end());
-                m_matcher->m_pattern_value_maps.erase(m_pattern_value_maps.begin() + m_capture_size,
-                                                      m_pattern_value_maps.end());
+                if (!m_matcher->m_matched_list.empty())
+                {
+                    m_matcher->m_matched_list.erase(m_matcher->m_matched_list.begin() + m_watermark,
+                                                    m_matcher->m_matched_list.end());
+                }
+
+                if (!m_pattern_value_maps.empty())
+                {
+                    m_matcher->m_pattern_value_maps.erase(
+                        m_pattern_value_maps.begin() + m_capture_size, m_pattern_value_maps.end());
+                }
+
                m_matcher->m_pattern_map = m_pattern_value_map;
            }
        }