code: def main(argv):
      if _SET_ENV.value:
          for env_str in _SET_ENV.value:
              # Stop matching at the first '=' since we want to capture
              # --set_env='FOO=--foo_a=1 --foo_b=2' all as part of FOO.
              env_list = env_str.split("=", 1)
              if len(env_list) == 2:
                  os.environ[env_list[0]] = env_list[1]
      benchmark.run_benchmarks()
  
file_path: jax/benchmarks/math_benchmark.py
imports: import functools
  import google_benchmark
  import jax
  import jax.numpy
  import numpy
  import os
  import sys
  from google_benchmark import Counter
  from absl import app
  from absl import flags
label_desc: Use image node_id 4 for calling a global function with example usage: main(argv) without return types