code: def __init__(
      self,
      code_string: str,
      return_by_ref: bool,
      num_outputs: int,
      **kwargs,
  ):
      self.code_string = code_string
  
      assert (
          return_by_ref or num_outputs == 1
      ), "Return by value only works for single output. "
      self.return_by_ref = return_by_ref
      self.num_outputs = num_outputs
  
      parsed_code = _CodeParser(code_string)
      self.kernel_name = parsed_code.function_name
  
      self.kwargs_dict = kwargs
      self.is_cuda_available = torch.cuda.is_available()
  
file_path: pytorch/torch/cuda/jiterator.py
imports: import re
  from typing import Callable, List
  import torch
  from torch import Tensor
label_desc: Use image node_id 1 to create a new _JittedFunction object  with example: obj = _JittedFunction(code_string, return_by_ref, num_outputs)