[BUG] Switch to rmm.to_device in RegressorMixin #1092
Labels
? - Needs Triage
Need team to review and classify
bug
Something isn't working
Cython / Python
Cython or Python issue
Recently we tried to use
rmm.to_device
inRegressorMixin
as part of the move to handle GPU array-like objects in PR ( #1086 ). However we encountered some issues usingrmm.to_device
with cuDF Series objects. ( rapidsai/rmm#129 ) As a result, we usednumba.to_device
instead, which didn't have this issue. Though ultimately we would like to switch to usingrmm.to_device
once that is viable.The text was updated successfully, but these errors were encountered: