MCPcopy Create free account
hub / github.com/PRIME-RL/PRIME / is_tensor_parallel_param

Function is_tensor_parallel_param

training/verl/utils/megatron/tensor_parallel.py:86–87  ·  view source on GitHub ↗
(param)

Source from the content-addressed store, hash-verified

84
85
86def is_tensor_parallel_param(param):
87 return (hasattr(param, 'tensor_model_parallel') and param.tensor_model_parallel)
88
89
90def get_tensor_parallel_partition_dim(param):

Callers 2

get_tensor_parallel_partition_dimFunction · 0.85
get_tensor_parallel_partition_strideFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected