Module simulated_annealing

Module simulated_annealing 

Source
Expand description

Repartitioning using simulated annealing algorithms.

Structs§

IntermediatePartitioningModel
A simulated annealing model that moves a random subtree to the partition that maximizes memory reduction.
LeafPartitioningModel
A simulated annealing model that moves a random tensor to the partition that maximizes memory reduction.
NaiveIntermediatePartitioningModel
A simulated annealing model that moves a random subtree between random partitions.
NaivePartitioningModel
A simulated annealing model that moves a random tensor between random partitions.
SimulatedAnnealingOptimizer
Optimizer that implements the simulated annealing algorithm

Traits§

OptModel
OptModel is a trait that defines requirements to be used with optimization algorithm

Functions§

balance_partitions
Runs simulated annealing to find a better partitioning.