class_docstr: Test Case: To test simple fork/wait operation in a graph
  fork is called on simple addition operation on input tensors
class_docstr_tok: ["Test","Case",":","To","test","simple","fork\/wait","operation","in","a","graph","fork","is","called","on","simple","addition","operation","on","input","tensors"]
code: def test_fork_wait_3_async(self):
      input = torch.ones(3, 3)
      num_forks = 10
      torch_graph = torch.jit.script(fork_wait_graph3)
      output_ref = torch_graph(input, num_forks)
      static_runtime_module = StaticModule(torch_graph)
      output_test = static_runtime_module.runAsync(
          (input, num_forks), {}
      )
      output_test.wait()
      torch.testing.assert_close(output_test.value(), output_ref)
  
imports: import unittest
  from typing import Dict, Optional
  import numpy
  import torch
  from torch import nn
  from torch.testing._internal.common_utils import TestCase, run_tests
  from typing import List
label_desc: Use image node_id 6 for calling the TestStaticModule obj's underlying member method code with example usage: obj.test_fork_wait_3_async() without return types