Setting up RMM allocator instead of CUPY

discovery-unicamp / dasf-core

Framework for computing Machine Learning algorithms in Python using Dask and RAPIDS AI.

MIT License

9 stars 2 forks source link

Closed otavioon closed 1 year ago

otavioon commented 1 year ago

Added support to RMM allocator

otavioon commented 1 year ago

Closed as already has gpu_allocator option