Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update cross-link to cuda-python object #1699

Merged
merged 2 commits into from
Oct 9, 2024
Merged

Update cross-link to cuda-python object #1699

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
2c2bf5c
Fix numpydoc section names
wence- Oct 9, 2024
530e03a
Update cross-link to cuda-python object
wence- Oct 9, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion python/rmm/rmm/pylibrmm/device_buffer.pyx
View file
Open in desktop
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
View file
Open in desktop
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
Loading