code: def function_2(i):
      i = i + 1
      return i
  
imports: import importlib.util
  import os
  import shutil
  import tempfile
  import pytest
  import torch
  import triton
  import triton.language
  from triton.runtime.jit import JITFunction
label_desc: Use image node_id 2 for calling a global function with example usage: function_2(i) and returns: i