torch.cuda.jiterator._create_multi_output_jit_fn
torch.cuda.jiterator._create_multi_output_jit_fn(code_string, num_outputs, **kwargs)
Create a jiterator-generated cuda kernel for an elementwise op that
   supports returning one or more outputs.
Parameters:
      * code_string (str) -- CUDA code string to be compiled by
        jiterator. The entry functor must return value by reference.
  * **num_outputs** (*int*) -- number of outputs return by the
    kernel

  * **kwargs** (*Dict**, **optional*) -- Keyword arguments for
    generated function

Return type:
      Callable
Example:
  code_string = "template <typename T> void my_kernel(T x, T y, T alpha, T& out) { out = -x + alpha * y; }"
  jitted_fn = create_jit_fn(code_string, alpha=1.0)
  a = torch.rand(3, device='cuda')
  b = torch.rand(3, device='cuda')
  # invoke jitted function like a regular python function
