code: def test_await_out_of_interpreter(self):
      def delayed(x: Tensor) -> Tensor:
          return 2 * (x + 1)
  
      def main(x: Tensor) -> Await[Tensor]:
          aw = torch.jit._awaitable(delayed, x)
          return aw
  
      inp = torch.eye(2)
  
      sm = torch.jit.script(main)
      out_aw = main(inp)
      out = torch.jit._awaitable_wait(out_aw)
  
      script_out_aw = sm(inp)
      script_out = torch.jit._awaitable_wait(script_out_aw)
      self.assertTrue(
          torch.allclose(
              2 * torch.eye(2) + 2 * torch.ones(2), script_out
          )
      )
      self.assertTrue(torch.allclose(script_out, out))
  
imports: import io
  import torch
  from torch.testing._internal.jit_utils import JitTestCase
  from torch.testing._internal.jit_utils import make_global
  from typing import List, Optional, Tuple
  from torch import Tensor
  from torch._awaits import _Await
label_desc: Use image node_id 14 for calling the TestAwait obj's underlying member method code with example usage: obj.test_await_out_of_interpreter() and returns: unknown, aw