Function gpu_topology

src/fused_mm_sampling/bench/sys_metadata.py:41–49

()

def gpu_topology() -> str | None:
    import subprocess

    if torch.cuda.device_count() < 2:
        return None
    try:
        return subprocess.check_output(["nvidia-smi", "topo", "-m"], text=True).strip()
    except (FileNotFoundError, subprocess.CalledProcessError):
        return "error"