Method prefetch_to_device

crates/cust/src/memory/unified.rs:673–687

Advises the driver to enqueue an operation on the stream to prefetch the memory to a certain GPU. This will cause the driver to fetch the data to the specified device as soon as the operation is reached on the stream. The device must have the attribute [`DeviceAttribute::ConcurrentManagedAccess`]. # Example ```no_run # fn main() -> Result<(), Box > { # let _context = cust:

(&self, stream: &Stream, device: &Device)

    /// # }
    /// ```
    fn prefetch_to_device(&self, stream: &Stream, device: &Device) -> CudaResult<()> {
        let slice = self.as_slice();
        let mem_size = std::mem::size_of_val(slice);

        unsafe {
            cuda::cuMemPrefetchAsync(
                slice.as_ptr() as cuda::CUdeviceptr,
                mem_size,
                device.as_raw(),
                stream.as_inner(),
            )
            .to_result()?;
        }
        Ok(())
    }

    /// Advises the driver that this memory range is mostly going to be read to, and occasionally written to.
    ///