code: def test_matmul_tuning_relay():
      def matmul(x, w):
          return torch.matmul(x, w)
  
      x = torch.randn(15, 20)
      w = torch.randn(20, 30)
      example_inputs = (x, w)
  
      rt_mod = optimize_torch(matmul, example_inputs)
      torch_answer = torch.matmul(x, w).numpy()
      tvm_answer = rt_mod(x, w).numpy()
  
      tvm.testing.assert_allclose(
          torch_answer, tvm_answer, atol=1e-5, rtol=1e-5
      )
  
imports: import tempfile
  import torch
  from torch.utils import benchmark
  from torchvision.models import resnet18
  import tvm
  import tvm.testing
  from tvm.contrib.torch import optimize_torch
  from tvm.meta_schedule import TuneConfig
label_desc: Use image node_id 1 for calling a global function with example usage: test_matmul_tuning_relay() and returns: torch