Struct DeviceBuffer 

pub struct DeviceBuffer<'g> { /* private fields */ }

显存里的一段 f32。借用创建它的 Gpu，保证在上下文释放前先释放。

Implementations

impl<'g> DeviceBuffer<'g>

pub fn len(&self) -> usize

元素个数。

pub fn is_empty(&self) -> bool

pub fn copy_from_host(&mut self, data: &[f32]) -> Result<(), String>

从内存拷入，长度必须相等。

pub fn copy_to_host(&self, out: &mut [f32]) -> Result<(), String>

拷回内存，长度必须相等。默认流上同步执行：之前提交的内核先跑完。

pub fn to_vec(&self) -> Result<Vec<f32>, String>

拷回成新的 Vec。

pub fn copy_from_device(&mut self, src: &DeviceBuffer<'_>) -> Result<(), String>

从另一段显存拷入，长度必须相等。

Trait Implementations

impl Drop for DeviceBuffer<'_>

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.