(self, proc_ranks)
| 53 | self.stream = _sh.get_next() |
| 54 | |
| 55 | def check(self, proc_ranks): |
| 56 | assert _sd is not None, "please call init_process_group first" |
| 57 | for rank in proc_ranks: |
| 58 | assert isinstance(rank, int) |
| 59 | assert rank >= 0 and rank < _sd.world_size |
| 60 | assert _sd.proc_rank in proc_ranks |
| 61 | |
| 62 | @property |
| 63 | def size(self): |