We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7720758 commit 0e65be3Copy full SHA for 0e65be3
1 file changed
docs/Out_of_core.md
@@ -29,7 +29,7 @@ from rmm.allocators.cupy import rmm_cupy_allocator
29
30
def set_mem_pool():
31
# Prefer pool allocator for performance and NVLink (managed memory can degrade P2P)
32
- rmm.reinitialize(managed_memory=False, pool_allocator=True)
+ rmm.reinitialize(managed_memory=True, pool_allocator=False)
33
cp.cuda.set_allocator(rmm_cupy_allocator)
34
35
client.run(set_mem_pool)
0 commit comments