Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/MotrixLab/ViMoGen
/ get_world_size
Function
get_world_size
parallel/utils.py:47–49 ·
view source on GitHub ↗
()
Source
from the content-addressed store, hash-verified
45
46
47
def
get_world_size():
48
# can be called before dist initialization
49
return
int(os.environ[
'WORLD_SIZE'
])
50
51
52
def
get_nodes():
Callers
1
get_device_mesh
Function · 0.85
Calls
no outgoing calls
Tested by
no test coverage detected