code: def run_parallel(all_testmodules, p):
      print("Running tests with parallelism=", p)
      available_gpu_tokens = list(range(p))
      executor = ThreadPoolExecutor(max_workers=p)
      # walking through test modules
      for testmodule in all_testmodules:
          executor.submit(run_test, testmodule, available_gpu_tokens)
      # waiting for all modules to finish
      executor.shutdown(wait=True)  # wait for all jobs to finish
      return
  
file_path: jax/build/rocm/run_single_gpu.py
imports: import argparse
  import os
  import re
  import subprocess
  import threading
  from concurrent.futures import ThreadPoolExecutor
label_desc: Use image node_id 4 for calling a global function with example usage: run_parallel(all_testmodules, p) without return types