code: def load(cls, device, so_path):
      runner = AOTIRunnerUtil.load_runner(device, so_path)
  
      def optimized(*args):
          call_spec = runner.get_call_spec()
          in_spec = pytree.treespec_loads(call_spec[0])
          out_spec = pytree.treespec_loads(call_spec[1])
          flat_inputs = fx_pytree.tree_flatten_spec(
              (*args, {}), in_spec
          )
          flat_outputs = runner.run(flat_inputs)
          return pytree.tree_unflatten(flat_outputs, out_spec)
  
      return optimized
  
file_path: pytorch/test/inductor/test_aot_inductor_utils.py
imports: import torch
  import torch._export
  import torch._inductor
  import torch.fx._pytree
  from torch.testing._internal.common_utils import IS_FBCODE
  from torch.utils import _pytree
label_desc: Use image node_id 3 for calling the AOTIRunnerUtil obj's underlying member method code with example usage: obj.load(cls, device, so_path) and returns: optimized, pytree