Function melior::dialect::ods::gpu::spmm_buffer_size

source ·

pub fn spmm_buffer_size<'c>(
    context: &'c Context,
    buffer_szs: &[Type<'c>],
    async_dependencies: &[Value<'c, '_>],
    spmat_a: Value<'c, '_>,
    dnmat_b: Value<'c, '_>,
    dnmat_c: Value<'c, '_>,
    compute_type: TypeAttribute<'c>,
    location: Location<'c>
) -> SpMMBufferSizeOperation<'c>

Expand description

Creates a spmm_buffer_size operation.