Skip to content

Commit 868c4cf

Browse files
authored
Cherry-pick "Remove PoolMemoryResource wrapper" into release (#7881)
Cherry-pick of #7878 into the release branch for 26.04. Authors: - Jim Crist-Harif (https://github.com/jcrist) Approvers: - Simon Adorf (https://github.com/csadorf) URL: #7881
1 parent 4c7ea21 commit 868c4cf

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)