(self, rank, world_size, backend=BACKEND)
| 168 | pass |
| 169 | |
| 170 | def dist_init(self, rank, world_size, backend=BACKEND): |
| 171 | store = dist.FileStore(self.file_name, world_size) |
| 172 | return dist.init_process_group( |
| 173 | backend=backend, |
| 174 | store=store, |
| 175 | rank=rank, |
| 176 | world_size=world_size |
| 177 | ) |
| 178 | |
| 179 | def construct_uneven_inputs(self, base, offset, device=None): |
| 180 | r""" |