Use cupy.array instead of librmm.device_array

rapidsai / deeplearning

337 stars 90 forks source link

Closed benfred closed 4 years ago

benfred commented 4 years ago

device_array is being removed from RMM https://github.com/rapidsai/rmm/issues/180 Switch to use a cupy for on device array's instead.