-
Notifications
You must be signed in to change notification settings - Fork 25.2k
Call the new _estimate_memory_usage API endpoint on df analytics _start #45536
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
Call the new _estimate_memory_usage API endpoint on df analytics _start #45536
Conversation
80402eb
to
66d6172
Compare
66d6172
to
2df1f74
Compare
Pinging @elastic/ml-core |
2df1f74
to
b5ec18f
Compare
run elasticsearch-ci/1 |
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.
Looks good! Left a few minor comments
.../src/main/java/org/elasticsearch/xpack/ml/action/TransportStartDataFrameAnalyticsAction.java
Show resolved
Hide resolved
.../src/main/java/org/elasticsearch/xpack/ml/action/TransportStartDataFrameAnalyticsAction.java
Outdated
Show resolved
Hide resolved
...n/java/org/elasticsearch/xpack/ml/dataframe/process/MemoryUsageEstimationProcessManager.java
Outdated
Show resolved
Hide resolved
.../src/main/java/org/elasticsearch/xpack/ml/action/TransportStartDataFrameAnalyticsAction.java
Outdated
Show resolved
Hide resolved
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.
LGTM
run elasticsearch-ci/1 |
2 similar comments
run elasticsearch-ci/1 |
run elasticsearch-ci/1 |
This PR integrates the new _estimate_memory_usage API endpoint into df analytics.
_estimate_memory_usage is called early during df analytics _start in order to detect whether user-provided model memory limit is sufficient taking into account memory usage estimation. If not, error is thrown and the user can retry the request with different (higher) value of model_memory_limit.
Relates #44699
Closes #45544