def METHOD_NAME(self) -> None:
    class Model(torch.nn.Module):
        def __init__(self):
            super().__init__()
            self.async_compute = AsyncModule()
        def forward(self, x: torch.Tensor) -> int:
            async_fut = self.async_compute(x)
            y = x * 3 + 5
            return async_fut.numel() + y.numel()
    m = Model()
    ref_res = m(torch.ones(3, 4))
    self.assertEqual(ref_res, 24)
    # ensure fx tracing works
    gm = torch.fx.symbolic_trace(m)
    traced_res = gm(torch.ones(3, 4))
    self.assertEqual(traced_res, ref_res)