-
Notifications
You must be signed in to change notification settings - Fork 24.9k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[ML] Add setting to scale the processor count used in the model assignment planner #98296
Conversation
Pinging @elastic/ml-core (Team:ML) |
Hi @davidkyle, I've created a changelog YAML for you. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Code changes look good, I'll test shortly when the docker image is up
@@ -63,7 +64,7 @@ class TrainedModelAssignmentRebalancer { | |||
this.deploymentToAdd = Objects.requireNonNull(deploymentToAdd); | |||
} | |||
|
|||
TrainedModelAssignmentMetadata.Builder rebalance() throws Exception { | |||
TrainedModelAssignmentMetadata.Builder rebalance(Settings settings) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
nit: Does it make sense to have Settings
be passed in the constructor of TrainedModelAssignmentRebalancer
and stored as a member to avoid passing it through a bunch of methods?
💔 Backport failed
You can use sqren/backport to manually backport by running |
…nment planner (elastic#98296) Adds the ml.allocated_processors_scale setting which is used to scale the value of ml.allocated_processors_double. This setting influences the number of model allocations that can fit on a node # Conflicts: # x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/MachineLearning.java # x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/autoscaling/MlAutoscalingResourceTracker.java # x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/utils/MlProcessors.java
…odel assignment planner (#98299) * [ML] Add setting to scale the processor count used in the model assignment planner (#98296) Adds the ml.allocated_processors_scale setting which is used to scale the value of ml.allocated_processors_double. This setting influences the number of model allocations that can fit on a node # Conflicts: # x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/MachineLearning.java # x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/autoscaling/MlAutoscalingResourceTracker.java # x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/utils/MlProcessors.java * non operator
Adds the
ml.allocated_processors_scale
setting which is used to scale the value ofml.allocated_processors_double
The bulk of the change is passing the
settings
object to where it is now needed, the only logic change is in MlProcessors.java where the returned processor count is scaled by this setting.