Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
[FEA] Allow DeviceNDArray as input data #369
when using cuml
gpu_data = rmm.to_device(np_data)
Allows methods in cuml to work directly with rmm.to_device would speed things along and, I believe, make for a nicer UX compared with:
We hadn't put the time before for this due to there not being a use case that pushed us into doing it (i.e. most gpu workflows so far have been cuDF into cuML). Now we have a good reasons to push for this in the next version.
That said, to properly use