code: def main(_):
      # CHECK-LABEL: TEST: neg int32[7]
      # CHECK: module @jit_neg
      # CHECK: func public @main
      print_ir(np.empty([7], np.int32))(lax.neg)
  
      # CHECK-LABEL: TEST: foo int32[7]
      # CHECK: module @jit_foo
      # CHECK: func public @main
      @print_ir(np.empty([7], np.int32))
      @jax.jit
      def foo(x):
          return x + 2
  
imports: from absl import app
  import jax
  from jax import lax
  import numpy
  from jax.tests.filecheck.jax_filecheck_helpers import print_ir
label_desc: Use image node_id 1 for calling a global function with example usage: main(_) and returns: unknown