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