MCPcopy
hub / github.com/Wan-Video/Wan2.2 / get_world_size

Function get_world_size

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

Source from the content-addressed store, hash-verified

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

Callers 12

__init__Method · 0.85
__init__Method · 0.85
__init__Method · 0.85
__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
all_to_allFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected