Struct melior::dialect::ods::gpu::AllocOperation

pub struct AllocOperation<'c> { /* private fields */ }

An alloc operation. GPU memory allocation operation..

The gpu.alloc operation allocates a region of memory on the GPU. It is similar to the memref.alloc op, but supports asynchronous GPU execution.

The op does not execute before all async dependencies have finished executing.

If the async keyword is present, the op is executed asynchronously (i.e. it does not block until the execution has finished on the device). In that case, it also returns a !gpu.async.token.

If the host_shared keyword is present, the memory will be allocated in a memory accessible both on host and on device.

Example:

%memref, %token = gpu.alloc async [%dep] host_shared (%width) : memref<64x?xf32, 1>

Implementations

impl<'c> AllocOperation<'c>

pub fn name() -> &'static str

Returns a name.

pub fn as_operation(&self) -> &Operation<'c>

Returns a generic operation.

pub fn builder( context: &'c Context, location: Location<'c> ) -> AllocOperationBuilder<'c, Unset, Unset, Unset, Unset>

Creates a builder.

pub fn memref(&self) -> Result<OperationResult<'c, '_>, Error>

pub fn async_token(&self) -> Result<OperationResult<'c, '_>, Error>

pub fn async_dependencies( &self ) -> Result<impl Iterator<Item = Value<'c, '_>>, Error>

pub fn dynamic_sizes( &self ) -> Result<impl Iterator<Item = Value<'c, '_>>, Error>

pub fn symbol_operands( &self ) -> Result<impl Iterator<Item = Value<'c, '_>>, Error>

pub fn host_shared(&self) -> Result<Attribute<'c>, Error>

pub fn set_host_shared(&mut self, value: Attribute<'c>)

pub fn remove_host_shared(&mut self) -> Result<(), Error>

Trait Implementations

impl<'c> From<AllocOperation<'c>> for Operation<'c>

fn from(operation: AllocOperation<'c>) -> Self

Converts to this type from the input type.

impl<'c> TryFrom<Operation<'c>> for AllocOperation<'c>

type Error = Error

The type returned in the event of a conversion error.

fn try_from(operation: Operation<'c>) -> Result<Self, Self::Error>

Performs the conversion.