communication_partitioning

Function communication_partitioning 

Source
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 partitions
  • partitioning_strategy - The strategy to pass to KaHyPar
  • min - if true performs min_cut to partition tensor network, if false, uses max_cut