Mojo struct
R2GOutputOp
struct R2GOutputOp[mut: Bool, //, accum_type: DType, dtype: DType, layout: Layout, num_threads: Int, mma_shape: IndexList[3], block_tile_shape: IndexList[3], o: Origin[mut]]
Fields
- c (
LayoutTensor[dtype, layout, o]):
Implemented traits
AnyType,
OutputOp,
UnknownDestructibility
Aliases
__del__is_trivial
alias __del__is_trivial = LayoutTensor[dtype, layout, o].__del__is_trivial
args_type
alias args_type = STOutputOpArgs[dtype, layout, o]
Methods
__init__
__init__(out self, args: STOutputOpArgs[dtype, layout, o])
to_kernel_args
static to_kernel_args(c: LayoutTensor[dtype, layout, o], ctx: DeviceContext) -> STOutputOpArgs[dtype, layout, o]
Returns:
__call__
__call__(self, tmem_addr: UInt32)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!