Multiple CPU interop fixes for serialization and cloning#6223
Multiple CPU interop fixes for serialization and cloning#6223rapids-bot[bot] merged 23 commits intorapidsai:branch-25.04from
Conversation
|
Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually. Contributors can view more details about this message here. |
|
/ok to test |
…get/set_params behave as expected matching wrapped estimator
|
/ok to test |
Co-authored-by: Simon Adorf <[email protected]>
|
|
||
|
|
||
| def test_kernel_ridge(): | ||
| import cupy as cp |
There was a problem hiding this comment.
Why move this here? Maybe we should leave a comment for people from the future to explain why it can't be imported at the top of the file (or move it back if this was just for debugging)
| km = cluster.KMeans(n_clusters=13) | ||
| ckm = cuml.KMeans.from_sklearn(km) | ||
|
|
||
| assert ckm.n_clusters == 13 |
There was a problem hiding this comment.
I think we should merge this PR. It improves things and fixes several things.
We can keep improving the from_/as_sklearn round tripping. I think the test from https://github.com/rapidsai/cuml/pull/6342/files#r1963552769 still doesn't pass (even if you exclude the raft handle). But lets look at that in a new PR
|
The two not optional jobs both fail with this: I added this test in #6327 where it passed. But, I am wondering if this test should always pass or only it he accelerator is enabled (it should be skipped if the accelerator is disabled)? When I wrote the test I meant "when the accelerator is active, it should always output a Numpy array". I didn't think about the "accelerator off" case, so happy to just skip this test in that case. I guess it depends on what the default output type is, mirror?, when the accelerator is off. The failures in the optional jobs look more serious. They are of this type: For several estimators :-/ There is also a CUDA error but maybe that is a spurious one? |
|
@betatim it was late last night, the error on the optional jobs is a small one, pushing a fix |
|
I'll merge this now. I think this is a good way forward to see what is what in combination with all the other PRs that are still open. There are things here that we could discuss/improve, let's do that in follow up PRs. |
|
/merge |
I think those comments have been addressed/we can come back to it
PRs being backported: - [x] #6234 - [x] #6306 - [x] #6320 - [x] #6319 - [x] #6327 - [x] #6333 - [x] #6142 - [x] #6223 - [x] #6235 - [x] #6317 - [x] #6331 - [x] #6326 - [x] #6332 - [x] #6347 - [x] #6348 - [x] #6337 - [x] #6355 - [x] #6354 - [x] #6322 - [x] #6353 - [x] #6359 - [x] #6364 - [x] #6363 - [x] [FIL BATCH_TREE_REORG fix for SM90, 100 and 120](a3e419a) --------- Co-authored-by: William Hicks <[email protected]>
No description provided.