def normalize_records()

in optimum_benchmark/profilers/ort_profiler.py [0:0]

• 20 lines of code
• 8 McCabe index (conditional complexity)

def normalize_records(data) -> List[Tuple[str, str, float]]:
    records = []
    for item in data:
        cat = item.get("cat")
        if cat is None:
            continue
        dur = item.get("dur")
        if dur is None:
            continue
        arg = item.get("args")
        if arg is None:
            continue
        op_name = arg.get("op_name")

        name = item["name"]

        if cat != "Kernel" and not name.endswith("kernel_time"):
            continue

        if cat in ["Kernel", "Node"]:
            LOGGER.debug(f"Kernel/Node {name} took {dur / 1e6:.2e} seconds")
            records.append((name.replace("_kernel_time", ""), op_name, dur / 1e6))

    return records