MCPcopy Index your code
hub / github.com/THUDM/slime / size

Method size

slime/utils/reloadable_process_group.py:192–193  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

190 return self.group.rank()
191
192 def size(self) -> int:
193 return self.group.size()
194
195 def name(self) -> str:
196 return self.group.name()

Callers 15

weight_dequantFunction · 0.80
fix_query_key_value_orderingMethod · 0.80
backwardMethod · 0.80
learnable_softmax_flash_attn_varlenFunction · 0.80
__init__Method · 0.80
forwardMethod · 0.80
forwardMethod · 0.80
get_absorb_query_key_value_tensorsMethod · 0.80
get_reinforce_plus_plus_returnsFunction · 0.80
calculate_log_probs_and_entropyFunction · 0.80
get_responsesFunction · 0.80
_allgather_cp_redistributeFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected