Skip to content

Conversation

@eordentlich
Copy link
Collaborator

@eordentlich eordentlich commented Oct 27, 2025

move stage-level scheduling invocation to _call_cuml_fit func so that knn and dbscan also benefit

these algos invoke cuml from the model and not the estimator _fit, so were not leveraging stage-level scheduling.

@eordentlich eordentlich requested a review from wbo4958 October 27, 2025 21:45
@eordentlich
Copy link
Collaborator Author

build

Copy link
Collaborator

@wbo4958 wbo4958 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@eordentlich eordentlich merged commit bf4ad1d into NVIDIA:main Oct 30, 2025
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants