🐛 Fix initializing the extractor by not removing phase nodes

DVLab-NTU · Dec 26, 2023 · fc29018 · fc29018
1 parent 449785c
commit fc29018
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/extractor/extract.cpp b/src/extractor/extract.cpp
@@ -771,7 +771,7 @@ void Extractor::update_neighbors() {
 
         if (num_boundaries != 2) continue;
 
-        if (_graph->get_num_neighbors(f) == 2) {
+        if (_graph->get_num_neighbors(f) == 2 && f->get_phase() == dvlab::Phase(0)) {
             // NOTE - Remove
             for (auto& [b, ep] : _graph->get_neighbors(f)) {
                 if (_graph->get_inputs().contains(b)) {