code: def test_conv3d():
      # test conv 3d
      data = torch.rand(2, 3, 4, 4, 4)
      conv3d = torch.nn.Conv3d(
          in_channels=3, out_channels=4, kernel_size=2
      )
      materialized_output = conv3d(data)
      _assert_output_shape(
          data=data,
          module=conv3d,
          patch_fn=patched_module.torch_nn_conv3d,
          expect_exception=False,
          output_shape=materialized_output.shape,
      )
  
      conv3d = torch.nn.Conv3d(
          in_channels=3, out_channels=4, kernel_size=2, padding=1
      )
      materialized_output = conv3d(data)
      _assert_output_shape(
          data=data,
          module=conv3d,
          patch_fn=patched_module.torch_nn_conv3d,
          expect_exception=False,
          output_shape=materialized_output.shape,
      )
  
      conv3d = torch.nn.Conv3d(
          in_channels=3,
          out_channels=4,
          kernel_size=2,
          padding=1,
          dilation=2,
      )
      materialized_output = conv3d(data)
      _assert_output_shape(
          data=data,
          module=conv3d,
          patch_fn=patched_module.torch_nn_conv3d,
          expect_exception=False,
          output_shape=materialized_output.shape,
      )
  
      conv3d = torch.nn.Conv3d(
          in_channels=3,
          out_channels=4,
          kernel_size=2,
          padding=1,
          dilation=2,
          padding_mode="reflect",
      )
      materialized_output = conv3d(data)
      _assert_output_shape(
          data=data,
          module=conv3d,
          patch_fn=patched_module.torch_nn_conv3d,
          expect_exception=False,
          output_shape=materialized_output.shape,
      )
  
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 8 for calling a global function with example usage: test_conv3d() without return types