MCPcopy
hub / github.com/Robbyant/lingbot-world / get_world_size

Function get_world_size

wan/distributed/util.py:16–17  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

14
15
16def get_world_size():
17 return dist.get_world_size()
18
19
20def all_to_all(x, scatter_dim, gather_dim, group=None, **kwargs):

Callers 11

__init__Method · 0.85
__init__Method · 0.85
forwardMethod · 0.85
forwardMethod · 0.85
after_transformer_blockMethod · 0.85
forwardMethod · 0.85
rope_applyFunction · 0.85
sp_dit_forwardFunction · 0.85
sp_dit_forward_causalFunction · 0.85
sp_attn_forward_causalFunction · 0.85
all_to_allFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected