does `get_device_for_address` (used in `src/neighbors/cagra.cuh`) work as intended on systems with AST or HMM?

[tfeher]

[julianmi]

I tested this on Grace Hopper and I see the expected behavior: raft::get_device_for_address(dataset.data_handle()) returns -1 for a dataset on the host and 0 for when it resides in device 0 memory.