Skip to content

Commit

Permalink
Update cross-link to cuda-python object (#1699)
Browse files Browse the repository at this point in the history
NVIDIA/cuda-python#137 reorganised the low-level binding structure which broke our cross-linking, update to the new name to fix.

- Closes #1698

Authors:
  - Lawrence Mitchell (https://github.com/wence-)

Approvers:
  - Matthew Murray (https://github.com/Matt711)
  - Bradley Dice (https://github.com/bdice)

URL: #1699
  • Loading branch information
wence- authored Oct 9, 2024
1 parent a381645 commit 4e519bb
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 11 deletions.
2 changes: 1 addition & 1 deletion python/rmm/rmm/pylibrmm/device_buffer.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -156,7 +156,7 @@ cdef class DeviceBuffer:
device : optional
The CUDA device to which to prefetch the memory for this buffer.
Defaults to the current CUDA device. To prefetch to the CPU, pass
:py:attr:`~cuda.cudart.cudaCpuDeviceId` as the device.
:py:attr:`~cuda.bindings.runtime.cudaCpuDeviceId` as the device.
stream : optional
CUDA stream to use for prefetching. Defaults to self.stream
"""
Expand Down
20 changes: 10 additions & 10 deletions python/rmm/rmm/statistics.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,8 +74,8 @@ def enable_statistics() -> None:
def get_statistics() -> Optional[Statistics]:
"""Get the current allocation statistics.
Return
------
Returns
-------
If enabled, returns the current tracked statistics.
If disabled, returns None.
"""
Expand All @@ -94,8 +94,8 @@ def push_statistics() -> Optional[Statistics]:
If statistics are disabled (the current memory resource is not an
instance of StatisticsResourceAdaptor), this function is a no-op.
Return
------
Returns
-------
If enabled, returns the current tracked statistics _before_ the pop.
If disabled, returns None.
"""
Expand All @@ -114,8 +114,8 @@ def pop_statistics() -> Optional[Statistics]:
If statistics are disabled (the current memory resource is not an
instance of StatisticsResourceAdaptor), this function is a no-op.
Return
------
Returns
-------
If enabled, returns the popped counters.
If disabled, returns None.
"""
Expand Down Expand Up @@ -232,8 +232,8 @@ def report(
ordered_by
Sort the statistics by this attribute.
Return
------
Returns
-------
The pretty formatted string of the memory statistics
"""

Expand Down Expand Up @@ -279,8 +279,8 @@ def _get_descriptive_name_of_object(obj: object) -> str:
obj
Object in question
Return
------
Returns
-------
A string including filename, line number, and object name.
"""

Expand Down

0 comments on commit 4e519bb

Please sign in to comment.