def METHOD_NAME(self, num_inputs):
    inputs = []
    for i in range(num_inputs):
        inputs.append(torch.rand(3, device='cuda').mul(10))
    input_string = ",".join([f"T i{i}" for i in range(num_inputs)])
    function_body = "+".join([f"i{i}" for i in range(num_inputs)])
    code_string = f"template <typename T> T my_kernel({input_string}) {{ return {function_body}; }}"
    jitted_fn = create_jit_fn(code_string)
    def ref_fn(*inputs):
        return torch.sum(torch.stack(inputs), dim=0)
    expected = ref_fn(*inputs)
    result = jitted_fn(*inputs)
    self.assertEqual(expected, result)