hub / github.com/OpenDriveLab/ReSim / get_context_parallel_world_size

Function get_context_parallel_world_size

sat/vae_modules/utils.py:49–52 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

47
48
49	def get_context_parallel_world_size():
50	assert _CONTEXT_PARALLEL_SIZE is not None, "context parallel size is not initialized"
51
52	return _CONTEXT_PARALLEL_SIZE
53
54
55	def get_context_parallel_rank():

_splitFunction · 0.90

_gatherFunction · 0.90

_conv_splitFunction · 0.90

_conv_gatherFunction · 0.90

_pass_from_previous_rankFunction · 0.90

_fake_cp_pass_from_previous_rankFunction · 0.90

forwardMethod · 0.90

no outgoing calls

no test coverage detected