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

Function get_context_parallel_group

sat/sgm/util.py:50–53  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

48
49
50def get_context_parallel_group():
51 assert _CONTEXT_PARALLEL_GROUP is not None, "context parallel group is not initialized"
52
53 return _CONTEXT_PARALLEL_GROUP
54
55
56def get_context_parallel_world_size():

Callers 6

_gatherFunction · 0.50
_conv_gatherFunction · 0.50
_pass_from_previous_rankFunction · 0.50
get_inputMethod · 0.50
encodeMethod · 0.50
decodeMethod · 0.50

Calls

no outgoing calls

Tested by

no test coverage detected