Function _is_torchrun

src/fused_mm_sampling/tp_info.py:60–62

Check if the current process was launched by torchrun.

()

def _is_torchrun() -> bool:
    """Check if the current process was launched by torchrun."""
    return "RANK" in os.environ and "WORLD_SIZE" in os.environ


def _torchrun_worker(fn: Callable, args: tuple) -> None: