code: def test_mm_plus_mm(self):
      def fn(a, b, c, d):
          return torch.add(torch.mm(a, b), torch.mm(c, d))
  
      args_list = [
          (
              torch.randn(16, 16, device="cuda"),
              torch.randn(16, 16, device="cuda"),
              torch.randn(16, 16, device="cuda"),
              torch.randn(16, 16, device="cuda"),
          ),
          # https://github.com/pytorch/pytorch/issues/100670.
          (
              torch.randn(1, 4, device="cuda"),
              torch.randn(4, 2, device="cuda"),
              torch.randn(1, 2, device="cuda"),
              torch.randn(2, 1, device="cuda"),
          ),
          (
              torch.randn(1, 2, device="cuda"),
              torch.randn(2, 1, device="cuda"),
              torch.randn(1, 4, device="cuda"),
              torch.randn(4, 2, device="cuda"),
          ),
          (
              torch.randn(1, 4, device="cuda"),
              torch.randn(4, 2, device="cuda"),
              torch.randn(1, 5, device="cuda"),
              torch.randn(5, 2, device="cuda"),
          ),
      ]
      for args in args_list:
          self.common(fn, args, 1, 3)
  
imports: import copy
  import unittest
  import torch
  import torch._dynamo.config
  import torch._inductor.config
  from torch._dynamo.test_case import run_tests, TestCase
  from torch._dynamo.utils import count_calls, counters
  from torch._higher_order_ops.out_dtype import out_dtype
  from torch._inductor.fx_passes import joint_graph
  from torch._inductor.fx_passes.serialized_patterns.central_index import get_serialized_pattern
  from torch._inductor.pattern_matcher import _TargetExpr, Arg, CallFunction, gen_pattern, KeywordArg, Match, PatternExpr, PatternMatcherPass, PatternPrettyPrinter, register_graph_pattern, stable_topological_sort
  from torch._inductor.utils import run_and_get_code
  from torch._inductor.virtualized import V
  from torch.testing import FileCheck
  from torch.testing._internal.common_cuda import SM80OrLater
  from torch.testing._internal.common_utils import IS_LINUX, skipIfRocm
  from torch.testing._internal.inductor_utils import HAS_CUDA
label_desc: Use image node_id 2 for calling the TestPatternMatcher obj's underlying member method code with example usage: obj.test_mm_plus_mm() and returns: torch