fix[venom]: fix invalid phis after SCCP

vyper/venom/passes/sccp/sccp.py

@@ -74,7 +74,11 @@ def run_pass(self):
 
         # self._propagate_variables()
 
-        self.analyses_cache.invalidate_analysis(CFGAnalysis)
+        if self.cfg_dirty:
+            self.analyses_cache.force_analysis(CFGAnalysis)
+            self._fix_phi_nodes()
+        else:
+            self.analyses_cache.invalidate_analysis(CFGAnalysis)
 
     def _calculate_sccp(self, entry: IRBasicBlock):
         """
@@ -329,6 +333,28 @@ def _replace_constants(self, inst: IRInstruction):
                 if isinstance(lat, IRLiteral):
                     inst.operands[i] = lat
 
+    def _fix_phi_nodes(self):
+        for bb in self.function.get_basic_blocks():
+            for inst in bb.instructions.copy():
+                if inst.opcode == "phi":
+                    self._fix_phi_node_inst(inst)
+
+    def _fix_phi_node_inst(self, phi_inst: IRInstruction):
+        if len(phi_inst.parent.cfg_in) != 1:
+            return
+
+        src_bb: IRBasicBlock = phi_inst.parent.cfg_in.first()
+        assert isinstance(src_bb, IRBasicBlock)
+
+        from_src_bb = filter(lambda x: x[0] == src_bb.label, phi_inst.phi_operands)
+        operands = list(map(lambda x: x[1], from_src_bb))
+
+        assert len(operands) == 1
+        assert isinstance(operands[0], IRVariable)
+        assert phi_inst.output is not None
+        phi_inst.output.value = operands[0]
+        phi_inst.parent.remove_instruction(phi_inst)
+
 
 def _meet(x: LatticeItem, y: LatticeItem) -> LatticeItem:
     if x == LatticeEnum.TOP:

vyper/venom/passes/simplify_cfg.py

@@ -11,24 +11,21 @@ class SimplifyCFGPass(IRPass):
     def _merge_blocks(self, a: IRBasicBlock, b: IRBasicBlock):
         a.instructions.pop()
         for inst in b.instructions:
-            assert inst.opcode != "phi", "Not implemented yet"
-            if inst.opcode == "phi":
-                a.instructions.insert(0, inst)
-            else:
-                inst.parent = a
-                a.instructions.append(inst)
+            assert inst.opcode != "phi", f"Instruction should never be phi {inst}"
+            inst.parent = a
+            a.instructions.append(inst)
 
         # Update CFG
         a.cfg_out = b.cfg_out
         if len(b.cfg_out) > 0:
-            next_bb = b.cfg_out.first()
-            next_bb.remove_cfg_in(b)
-            next_bb.add_cfg_in(a)
-
-            for inst in next_bb.instructions:
-                if inst.opcode != "phi":
-                    break
-                inst.operands[inst.operands.index(b.label)] = a.label
+            for next_bb in b.cfg_out:
+                next_bb.remove_cfg_in(b)
+                next_bb.add_cfg_in(a)
+
+                for inst in next_bb.instructions:
+                    if inst.opcode != "phi":
+                        break
+                    inst.operands[inst.operands.index(b.label)] = a.label
 
         self.function.remove_basic_block(b)