Core JIT coroutine implementation

Summary:
Adds our custom version of the coroutine types for JIT.

Coroutines are mostly the same as generators in Python, with mostly superficial differences. Particularly they raise slightly differently worded errors or new errors in places where it doesn't make sense to use the languge-level syntax.

As with the generator implementation, we reuse most things from the core Python coroutine implemenation.

Reviewed By: DinoV

Differential Revision: D67425130

fbshipit-source-id: 8734adc88976d67612323d2cf504b0d53387a339

Lib/test/test_coroutines.py

@@ -36,7 +36,8 @@ async def asynciter(iterable):
 
 
 def run_async(coro):
-    assert coro.__class__ in {types.GeneratorType, types.CoroutineType}
+    # TODO(T209747648): Find an upstreamable way to fix this
+    # assert coro.__class__ in {types.GeneratorType, types.CoroutineType}
 
     buffer = []
     result = None
@@ -50,7 +51,8 @@ def run_async(coro):
 
 
 def run_async__await__(coro):
-    assert coro.__class__ is types.CoroutineType
+    # TODO(T209747648): Find an upstreamable way to fix this
+    # assert coro.__class__ is types.CoroutineType
     aw = coro.__await__()
     buffer = []
     result = None
@@ -530,7 +532,8 @@ async def foo():
             return 10
 
         f = foo()
-        self.assertIsInstance(f, types.CoroutineType)
+        # TODO(T209747648): Find an upstreamable way to fix this
+        # self.assertIsInstance(f, types.CoroutineType)
         self.assertTrue(bool(foo.__code__.co_flags & inspect.CO_COROUTINE))
         self.assertFalse(bool(foo.__code__.co_flags & inspect.CO_GENERATOR))
         self.assertTrue(bool(f.cr_code.co_flags & inspect.CO_COROUTINE))
@@ -1124,7 +1127,8 @@ def test_await_14(self):
         class Wrapper:
             # Forces the interpreter to use CoroutineType.__await__
             def __init__(self, coro):
-                assert coro.__class__ is types.CoroutineType
+                # TODO(T209747648): Find an upstreamable way to fix this
+                # assert coro.__class__ is types.CoroutineType
                 self.coro = coro
             def __await__(self):
                 return self.coro.__await__()

Lib/test/test_generators.py

@@ -206,21 +206,19 @@ def __del__(self):
         finally:
             gc.set_threshold(*thresholds)
 
-    # TODO(T209747648) - These should just be enabled as coroutines and
-    # async-generators work.
-    #
+    # TODO(T209747648) - This should be enabled when async-generators work.
     # def test_ag_frame_f_back(self):
     #     async def f():
     #         yield
     #     ag = f()
     #     self.assertIsNone(ag.ag_frame.f_back)
-    #
-    # def test_cr_frame_f_back(self):
-    #     async def f():
-    #         pass
-    #     cr = f()
-    #     self.assertIsNone(cr.cr_frame.f_back)
-    #     cr.close()  # Suppress RuntimeWarning.
+
+    def test_cr_frame_f_back(self):
+        async def f():
+            pass
+        cr = f()
+        self.assertIsNone(cr.cr_frame.f_back)
+        cr.close()  # Suppress RuntimeWarning.
 
     def test_gi_frame_f_back(self):
         def f():