Commit 63a417ac authored by Dmitry I. Lyakh's avatar Dmitry I. Lyakh

Bug fix in Metis optimizer + default parameter change.

parent 17bd0bd4
Pipeline #104897 passed with stage
in 13 minutes and 43 seconds
......@@ -69,7 +69,7 @@ double ContractionSeqOptimizerMetis::determineContractionSequence(const TensorNe
auto rnd = std::bind(distribution,generator);
double max_flop = 0.0;
partition_granularity_ = std::min(partition_granularity_,num_tensors/(2*partition_max_size_));
partition_granularity_ = std::max(partition_factor_,std::min(partition_granularity_,num_tensors/(2*partition_max_size_)));
while(partition_granularity_ >= partition_factor_){
auto num_walkers = num_walkers_;
while(num_walkers-- > 0){
......
......@@ -48,7 +48,7 @@ protected:
static constexpr const std::size_t PARTITION_FACTOR = 2;
static constexpr const std::size_t PARTITION_MAX_SIZE = 3;
static constexpr const std::size_t PARTITION_IMBALANCE_DEPTH = 24;
static constexpr const std::size_t PARTITION_IMBALANCE_DEPTH = 8;
static constexpr const std::size_t PARTITION_GRANULARITY = PARTITION_IMBALANCE_DEPTH;
static constexpr const double PARTITION_IMBALANCE = 1.3;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment