INNER CODE UNIT · Python

main

veg/hyphy · contrib/slurm_cpu_detect.py:51

def main():
    parser = argparse.ArgumentParser(description="Detect common CPU flags across SLURM nodes.")
    parser.add_argument("--cmake", action="store_true", help="Output in CMake -D format")
    args = parser.parse_args()

    nodes = get_slurm_nodes()
    if not nodes:
        if not args.cmake:
            print("No SLURM nodes found or sinfo not available.")
        sys.exit(0)

    node_flags = get_node_flags(nodes)
    if not node_flags:
        if not args.cmake:
            print("Failed to retrieve CPU flags from nodes.")
        sys.exit(1)

    common = find_common_flags(node_flags)

View source record →