Add code generator for yield stmt

diff --git a/Lib/compiler/pycodegen.py b/Lib/compiler/pycodegen.py
index bfb50597b9aa71e86afbdd16b691584a033a6355..0097482447c344ffc7c421a0a5192efb2697b322 100644 (file)
--- a/Lib/compiler/pycodegen.py
+++ b/Lib/compiler/pycodegen.py
@@ -909,6 +909,11 @@ class CodeGenerator:
         self.visit(node.value)
         self.emit('RETURN_VALUE')
 
+    def visitYield(self, node):
+        self.set_lineno(node)
+        self.visit(node.value)
+        self.emit('YIELD_STMT')
+
     # slice and subscript stuff
 
     def visitSlice(self, node, aug_flag=None):

diff --git a/Tools/compiler/compiler/pycodegen.py b/Tools/compiler/compiler/pycodegen.py
index bfb50597b9aa71e86afbdd16b691584a033a6355..0097482447c344ffc7c421a0a5192efb2697b322 100644 (file)
--- a/Tools/compiler/compiler/pycodegen.py
+++ b/Tools/compiler/compiler/pycodegen.py
@@ -909,6 +909,11 @@ class CodeGenerator:
         self.visit(node.value)
         self.emit('RETURN_VALUE')
 
+    def visitYield(self, node):
+        self.set_lineno(node)
+        self.visit(node.value)
+        self.emit('YIELD_STMT')
+
     # slice and subscript stuff
 
     def visitSlice(self, node, aug_flag=None):