Method is_rank0

src/fused_mm_sampling/tp_info.py:20–21

(self)

    def is_rank0(self) -> bool:
        return self.rank == 0

    def rank0_print(self, *args, **kwargs) -> None:
        if self.is_rank0():
            print(*args, **kwargs)