mirror of
https://github.com/Lizonghang/prima.cpp.git
synced 2025-09-09 15:44:35 +00:00
Merge branch 'speculative' into dev
This commit is contained in:
commit
fbbc30c950
6 changed files with 104 additions and 41 deletions
|
@ -1248,6 +1248,10 @@ static bool assign_layers_to_device(
|
|||
return cost * k;
|
||||
}
|
||||
);
|
||||
// apply higher priority to the head device, here 0.99 is a heuristic value
|
||||
// to ensure that small models in homogeneous clusters result in 32:0 partitioning,
|
||||
// rather than 1:31.
|
||||
model.lp_.col_cost_[0] *= 0.99;
|
||||
|
||||
// define the variable bounds
|
||||
model.lp_.col_lower_ = std::vector<double>(n_world * 2, 0.0);
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue