Function _print_numa_info

benchmarking/nsys_wrapper.py:79–88

Print NUMA/CPU topology before nsys captures stdout.

(rank: str)

def _print_numa_info(rank: str) -> None:
    """Print NUMA/CPU topology before nsys captures stdout."""
    cpus = sorted(os.sched_getaffinity(0))
    print(f"[rank {rank}] CPUs (after bind): {cpus}", flush=True)
    if rank == "0":
        try:
            topo = subprocess.check_output(["nvidia-smi", "topo", "-m"], text=True).strip()
            print(f"GPU topology:\n{topo}", flush=True)
        except (FileNotFoundError, subprocess.CalledProcessError) as e:
            print(f"[rank {rank}] Failed to get GPU topology: {e}", flush=True)


def main():