Handle GET_YIELD_FROM_ITER and END_SEND correctly

`GET_YIELD_FROM_ITER` doesn't consume its operand if its a coroutine
or generator, so treat it conservatively.

`END_SEND` leaves the TOS.

Author: mpage

Lib/test/test_peepholer.py

@@ -2663,6 +2663,24 @@ def test_set_function_attribute(self):
         ]
         self.cfg_optimization_test(insts, expected, consts=[None])
 
+    def test_get_yield_from_iter(self):
+        # GET_YIELD_FROM_ITER may leave its operand on the stack
+        insts = [
+            ("LOAD_FAST", 0, 1),
+            ("GET_YIELD_FROM_ITER", None, 2),
+            ("LOAD_CONST", 0, 3),
+            send := self.Label(),
+            ("SEND", end := self.Label(), 5),
+            ("YIELD_VALUE", 1, 6),
+            ("RESUME", 2, 7),
+            ("JUMP", send, 8),
+            end,
+            ("END_SEND", None, 9),
+            ("LOAD_CONST", 0, 10),
+            ("RETURN_VALUE", None, 11),
+        ]
+        self.cfg_optimization_test(insts, insts, consts=[None])
+
     def test_del_in_finally(self):
         # This loads `obj` onto the stack, executes `del obj`, then returns the
         # `obj` from the stack. See gh-133371 for more details.

Python/flowgraph.c

@@ -2874,6 +2874,7 @@ optimize_load_fast(cfg_builder *g)
                 case GET_ANEXT:
                 case GET_ITER:
                 case GET_LEN:
+                case GET_YIELD_FROM_ITER:
                 case IMPORT_FROM:
                 case MATCH_KEYS:
                 case MATCH_MAPPING:
@@ -2908,6 +2909,16 @@ optimize_load_fast(cfg_builder *g)
                     break;
                 }
 
+                case END_SEND:
+                case SET_FUNCTION_ATTRIBUTE: {
+                    assert(_PyOpcode_num_popped(opcode, oparg) == 2);
+                    assert(_PyOpcode_num_pushed(opcode, oparg) == 1);
+                    ref tos = ref_stack_pop(&refs);
+                    ref_stack_pop(&refs);
+                    PUSH_REF(tos.instr, tos.local);
+                    break;
+                }
+
                 // Opcodes that consume some inputs and push new values
                 case CHECK_EXC_MATCH: {
                     ref_stack_pop(&refs);
@@ -2944,16 +2955,6 @@ optimize_load_fast(cfg_builder *g)
                     break;
                 }
 
-                case SET_FUNCTION_ATTRIBUTE: {
-                    assert(_PyOpcode_num_popped(opcode, oparg) == 2);
-                    assert(_PyOpcode_num_pushed(opcode, oparg) == 1);
-                    ref func = ref_stack_pop(&refs);
-                    // Pop attr
-                    ref_stack_pop(&refs);
-                    PUSH_REF(func.instr, func.local);
-                    break;
-                }
-
                 // Opcodes that consume all of their inputs
                 default: {
                     int num_popped = _PyOpcode_num_popped(opcode, oparg);