Struct CopyPlan 

pub struct CopyPlan { /* private fields */ }

Implementations

impl CopyPlan

pub fn create( context: &Context, desc: &CopyDescriptor, workspace: Workspace, ) -> Result<Self>

pub unsafe fn from_raw( handle: cutensorMgCopyPlan_t, context: &Context, dst_pointer_count: usize, src_pointer_count: usize, workspace: Workspace, ) -> Result<Self>

Wraps an existing cuTENSORMg copy plan.

Safety

handle must be a valid cuTENSORMg copy plan associated with context and workspace. Pointer counts must match the descriptors used to create the plan. The returned value takes ownership of handle and destroys it on drop.

pub fn workspace(&self) -> &Workspace

pub unsafe fn copy_raw( &self, dst: &mut [*mut ()], src: &[*const ()], device_workspace: &mut [*mut ()], host_workspace: *mut (), streams: &[StreamBinding], ) -> Result<()>

Executes this copy plan with raw distributed tensor pointers.

Safety

The pointer arrays must match the tensor descriptors used to create the plan. Device workspace pointers and streams must be ordered like the MG context devices.

pub const fn as_raw(&self) -> cutensorMgCopyPlan_t

pub fn into_raw(self) -> cutensorMgCopyPlan_t

Consumes this plan and returns the owned raw cuTENSORMg copy plan.

The caller becomes responsible for destroying the plan.

Trait Implementations

impl Debug for CopyPlan

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Drop for CopyPlan

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.