Skip to content

Commit 3d93ae9

Browse files
committed
Remove PoolMemoryResource wrapper
I'm seeing some OOMs in CI after the `PoolMemoryResource` was added. Trying removing it to see if that fixes it.
1 parent 8c06f91 commit 3d93ae9

1 file changed

Lines changed: 3 additions & 9 deletions

File tree

‎python/cuml/cuml/accel/core.py‎

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -186,12 +186,8 @@ def install(
186186
import rmm
187187

188188
mr = rmm.mr.get_current_device_resource()
189-
if (
190-
isinstance(mr, rmm.mr.PrefetchResourceAdaptor)
191-
and isinstance(mr.upstream_mr, rmm.mr.PoolMemoryResource)
192-
and isinstance(
193-
mr.upstream_mr.upstream_mr, rmm.mr.ManagedMemoryResource
194-
)
189+
if isinstance(mr, rmm.mr.PrefetchResourceAdaptor) and isinstance(
190+
mr.upstream_mr, rmm.mr.ManagedMemoryResource
195191
):
196192
# Nothing to do
197193
pass
@@ -203,9 +199,7 @@ def install(
203199
else:
204200
rmm.mr.set_current_device_resource(
205201
rmm.mr.PrefetchResourceAdaptor(
206-
rmm.mr.PoolMemoryResource(
207-
rmm.mr.ManagedMemoryResource()
208-
)
202+
rmm.mr.ManagedMemoryResource()
209203
)
210204
)
211205
logger.debug("Enabled managed memory.")

0 commit comments

Comments
 (0)