pub fn communication_partitioning(
tensors: &[(usize, Tensor)],
k: i32,
imbalance: f64,
partitioning_strategy: PartitioningStrategy,
min: bool,
) -> Vec<usize>Expand description
Repeatedly partitions a tensor network to identify a communication scheme.
Returns a Vec<usize> of length equal to the number of input tensors minus one, acts as a communication scheme.
ยงArguments
tensors- &[(usize,Tensor)] to be partitioned. each tuple contains the intermediate contraction cost and intermediate tensor for communication.k- number of partitionspartitioning_strategy- The strategy to pass toKaHyParmin- iftrueperformsmin_cutto partition tensor network, iffalse, usesmax_cut