code: def test_linear():
      # test linear patch can produce the meta output with correct shape
      data = torch.rand(2, 4, device="meta")
      module = torch.nn.Linear(4, 2)
      _assert_output_shape(
          data,
          module,
          patched_module.torch_nn_linear,
          False,
          torch.Size([2, 2]),
      )
  
      # test if the linear patch can catch exception when dimension does not match
      data = torch.rand(2, 2, device="meta")
      _assert_output_shape(
          data, module, patched_module.torch_nn_linear, True, None
      )
  
file_path: ColossalAI/tests/test_fx/test_tracer/test_patched_module.py
imports: import torch
  from colossalai.fx.tracer.meta_patch import patched_module
  from colossalai.testing import clear_cache_before_run
label_desc: Use image node_id 3 for calling a global function with example usage: test_linear() without return types