Method from_world

src/fused_mm_sampling/tp_info.py:28–29

(cls)

    @classmethod
    def from_world(cls) -> "TPInfo":
        return cls(rank=dist.get_rank(), size=dist.get_world_size())


TP1 = TPInfo(rank=0, size=1)