From 070f5488882a97e37bd619e463256a0317cba908 Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Thu, 1 Aug 2024 08:14:08 -0500 Subject: [PATCH 1/2] Fix docs cross reference in DeviceBuffer.prefetch --- python/rmm/rmm/_lib/device_buffer.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/rmm/rmm/_lib/device_buffer.pyx b/python/rmm/rmm/_lib/device_buffer.pyx index b539881e7..13ba3a4f6 100644 --- a/python/rmm/rmm/_lib/device_buffer.pyx +++ b/python/rmm/rmm/_lib/device_buffer.pyx @@ -149,7 +149,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 - `~cuda.cudart.cudaCpuDeviceId` as the device. + :py:`~cuda.cudart.cudaCpuDeviceId` as the device. stream : optional CUDA stream to use for prefetching. Defaults to self.stream """ From d14d101d893ccde4a9c708dbd353d6f56eba470d Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Fri, 2 Aug 2024 17:02:38 -0500 Subject: [PATCH 2/2] Use :py:attr: Co-authored-by: Lawrence Mitchell --- python/rmm/rmm/_lib/device_buffer.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/rmm/rmm/_lib/device_buffer.pyx b/python/rmm/rmm/_lib/device_buffer.pyx index 13ba3a4f6..94a4dc771 100644 --- a/python/rmm/rmm/_lib/device_buffer.pyx +++ b/python/rmm/rmm/_lib/device_buffer.pyx @@ -149,7 +149,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:`~cuda.cudart.cudaCpuDeviceId` as the device. + :py:attr:`~cuda.cudart.cudaCpuDeviceId` as the device. stream : optional CUDA stream to use for prefetching. Defaults to self.stream """