docstr: Tests that the addmm meta kernel returns the correct output type
docstr_tok: ["Tests","that","the","addmm","meta","kernel","returns","the","correct","output","type"]
code: def test_addmm(self):
      """Tests that the addmm meta kernel returns the correct output type"""
      input = torch.ones(2, 2, dtype=torch.float16).to("lazy")
      self.assertEqual(input.dtype, torch.float16)
  
      fc_nobias = torch.nn.Linear(2, 2, bias=False, dtype=float16).to(
          "lazy"
      )
      out_nobias = fc_nobias(input)
      self.assertEqual(out_nobias.dtype, torch.float16)
  
      fc_bias = torch.nn.Linear(2, 2, bias=True, dtype=float16).to(
          "lazy"
      )
      out_bias = fc_bias(input)
      self.assertEqual(out_bias.dtype, torch.float16)
  
file_path: pytorch/test/lazy/test_meta_kernel.py
imports: import torch
  from torch.testing._internal.common_utils import TestCase
  from torch import float32, float16
  import torch._lazy
  import torch._lazy.ts_backend
label_desc: Use image node_id 2 for calling the TestMetaKernel obj's underlying member method code with example usage: obj.test_addmm() without return types