rapidsai / rmm

RAPIDS Memory Manager
https://docs.rapids.ai/api/rmm/stable/
Apache License 2.0
446 stars 188 forks source link

[FEA] CI should support HMM or Grace Hopper #1599

Open rongou opened 1 week ago

rongou commented 1 week ago

Is your feature request related to a problem? Please describe. We now have a system_memory_resource that allows system-allocated memory for device use. However, this only works on HMM or Grace Hopper, which the current CI pipelines don't support, so most of its tests are skipped.

Describe the solution you'd like We need CI that supports HMM or GH. Probably the easiest way is to have a Turing GPU that runs open-source drivers (see https://developer.nvidia.com/blog/simplifying-gpu-application-development-with-heterogeneous-memory-management/).

Describe alternatives you've considered If we could get a Grace Hopper machine for CI, that would be even better, but it's probably not very available in the near future.

Additional context PR that added the memory resource: #1581

jakirkham commented 1 week ago

cc @ajschmidt8