Function _print_gpu_topology

src/fused_mm_sampling/tp_info.py:110–115

()

def _print_gpu_topology() -> None:
    try:
        topo = subprocess.check_output(["nvidia-smi", "topo", "-m"], text=True).strip()
        print(f"GPU topology:\n{topo}")
    except (FileNotFoundError, subprocess.CalledProcessError) as e:
        warnings.warn(f"Failed to get GPU topology: {e}", stacklevel=2)


def _numa_bind(rank: int) -> None: