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_trivial_graph(self):
      s = torch.full((2, 2), 2)
      tg = torch.jit.script(trivial_graph)
      o_ref = tg(s, s, s)
      torch._C._fuse_to_static_module(tg.graph)
      assert "StaticSubgraph" in str(tg.graph)
      o_test = tg(s, s, s)
      torch.testing.assert_close(o_ref, o_test)
  
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 17 for calling the TestStaticModule obj's underlying member method code with example usage: obj.test_fusion_trivial_graph() without return types