Struct melior::dialect::ods::gpu::SpMVBufferSizeOperation

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

A spmv_buffer_size operation. Precompute buffersize for SpMV operation.

The gpu.spmv_buffer_size operation returns the buffer size required to perform the SpMV operation on the given sparse matrix and dense vectors. The operation expects handles returned by previous sparse operations to construct an environment and the operands for SpMV.

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 returns a !gpu.async.token in addition to the environment.

The matrix arguments can also be associated with one of the following operators: NON_TRANSPOSE, TRANSPOSE, CONJUGATE_TRANSPOSE. The default value is NON_TRANSPOSE.

Example:

%buffersz, %token = gpu.spmv_buffer_size async [%dep] %spmatA{TRANSPOSE}, %dnX, %dnY into f32

Implementations

impl<'c> SpMVBufferSizeOperation<'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> ) -> SpMVBufferSizeOperationBuilder<'c, Unset, Unset, Unset, Unset, Unset, Unset>

Creates a builder.

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

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

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

pub fn spmat_a(&self) -> Result<Value<'c, '_>, Error>

pub fn dn_x(&self) -> Result<Value<'c, '_>, Error>

pub fn dn_y(&self) -> Result<Value<'c, '_>, Error>

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

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

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

pub fn compute_type(&self) -> Result<TypeAttribute<'c>, Error>

pub fn set_compute_type(&mut self, value: TypeAttribute<'c>)

Trait Implementations

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

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

Converts to this type from the input type.

impl<'c> TryFrom<Operation<'c>> for SpMVBufferSizeOperation<'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.