code: def test_save_load_function():
      def foo(x):
          return 2 * x + 1
  
      example_input = torch.rand(3)
      opt_foo = optimize_torch(foo, example_input)
      ret1 = opt_foo(example_input)
      with tempfile.NamedTemporaryFile(suffix=".pt") as tmp:
          torch.save(opt_foo, tmp.name)
          loaded_mod = torch.load(tmp.name)
          ret2 = loaded_mod(example_input)
      tvm.testing.assert_allclose(
          ret1.numpy(), ret2.numpy(), 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 3 for calling a global function with example usage: test_save_load_function() and returns: unknown