Function get_context_parallel_group

sat/vae_modules/utils.py:43–46 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

41
42
43	def get_context_parallel_group():
44	assert _CONTEXT_PARALLEL_GROUP is not None, "context parallel group is not initialized"
45
46	return _CONTEXT_PARALLEL_GROUP
47
48
49	def get_context_parallel_world_size():

_gatherFunction · 0.90

_conv_gatherFunction · 0.90

_pass_from_previous_rankFunction · 0.90

_fake_cp_pass_from_previous_rankFunction · 0.90

forwardMethod · 0.90

get_inputMethod · 0.90

encodeMethod · 0.90

decodeMethod · 0.90

no outgoing calls

no test coverage detected