Function find_and_process_ttgir

benchmarking/insert_proton_records.py:226–254

Find all FMMS kernel TTGIR files in the dump directory and process them.

(dump_dir: str)

def find_and_process_ttgir(dump_dir: str) -> None:
    """Find all FMMS kernel TTGIR files in the dump directory and process them."""
    if not os.path.isdir(dump_dir):
        print(f"Error: directory not found: {dump_dir}", file=sys.stderr)
        sys.exit(1)

    ttgir_files = glob.glob(os.path.join(dump_dir, "**", "*.ttgir"), recursive=True)
    if not ttgir_files:
        print(f"No .ttgir files found in {dump_dir}", file=sys.stderr)
        sys.exit(1)

    fmms_files = [f for f in ttgir_files if "fused_mm_sample" in os.path.basename(f)]
    if not fmms_files:
        print(
            f"No fused_mm_sample*.ttgir files found in {dump_dir}. "
            f"Found: {[os.path.basename(f) for f in ttgir_files]}",
            file=sys.stderr,
        )
        sys.exit(1)

    for ttgir_file in fmms_files:
        try:
            print(f"Processing {ttgir_file}...")
            add_proton_records(ttgir_file)
        except AssertionError as e:
            print(f"Skipping {ttgir_file}: {e}")
        except Exception as e:
            print(f"Error processing {ttgir_file}: {e}")
            raise


if __name__ == "__main__":