Get the data parallel group the caller rank belongs to.
()
| 114 | |
| 115 | |
| 116 | def get_data_parallel_group(): |
| 117 | """Get the data parallel group the caller rank belongs to.""" |
| 118 | assert _DATA_PARALLEL_GROUP is not None, \ |
| 119 | 'data parallel group is not initialized' |
| 120 | return _DATA_PARALLEL_GROUP |
| 121 | |
| 122 | |
| 123 | def get_node_group(): |
no outgoing calls
no test coverage detected