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_fusion_outputs(self):
      a = torch.randn(2, 2)
      b = torch.randn(2, 2)
      c = 4
      og = torch.jit.script(output_graph)
      o_ref = og(a, b, b, c)
      torch._C._fuse_to_static_module(og.graph)
      assert "StaticSubgraph" in str(og.graph)
      o_test = og(a, b, b, c)
      for i in o_ref.keys():
          torch.testing.assert_close(o_ref[i], o_test[i])
  
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 20 for calling the TestStaticModule obj's underlying member method code with example usage: obj.test_fusion_outputs() without return types