Skip to content

Commit

Permalink
Update RMM imports to use pylibrmm (#283)
Browse files Browse the repository at this point in the history
This PR updates the RMM import to use `pylibrmm` now that `rmm._lib` is deprecated . It should be merged after rapidsai/rmm#1676.

Authors:
  - Matthew Murray (https://github.com/Matt711)

Approvers:
  - Peter Andreas Entschev (https://github.com/pentschev)

URL: #283
  • Loading branch information
Matt711 authored Oct 7, 2024
1 parent 7cdaa6f commit 0d42882
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/ucxx/ucxx/_lib/libucxx.pyx
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# SPDX-FileCopyrightText: Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES.
# SPDX-FileCopyrightText: Copyright (c) 2022-2024, NVIDIA CORPORATION & AFFILIATES.
# SPDX-License-Identifier: BSD-3-Clause


Expand Down Expand Up @@ -31,7 +31,7 @@ from libcpp.vector cimport vector

import numpy as np

from rmm._lib.device_buffer cimport DeviceBuffer
from rmm.pylibrmm.device_buffer cimport DeviceBuffer

from .arr cimport Array
from .ucxx_api cimport *
Expand Down

0 comments on commit 0d42882

Please sign in to comment.