MCPcopy Create free account
hub / github.com/OpenDriveLab/ReSim / get_node_rank

Function get_node_rank

SwissArmyTransformer/sat/mpu/initialize.py:145–147  ·  view source on GitHub ↗

Return my rank for the node group.

()

Source from the content-addressed store, hash-verified

143
144
145def get_node_rank():
146 """Return my rank for the node group."""
147 return torch.distributed.get_rank(group=get_node_group())
148
149
150def get_model_parallel_src_rank():

Callers 2

from_pretrainedMethod · 0.90
from_pretrainedMethod · 0.90

Calls 1

get_node_groupFunction · 0.85

Tested by

no test coverage detected