Mojo struct
R2GOutputOp
struct R2GOutputOp[accum_type: DType, dtype: DType, layout: Layout, num_threads: Int, mma_shape: IndexList[3], block_tile_shape: IndexList[3], o: MutOrigin]
Fields
- c (
LayoutTensor[dtype, layout, o]):
Implemented traits
AnyType,
Copyable,
DevicePassable,
ImplicitlyCopyable,
Movable,
OutputOp,
UnknownDestructibility
comptime members
__copyinit__is_trivial
comptime __copyinit__is_trivial = True
__del__is_trivial
comptime __del__is_trivial = True
__moveinit__is_trivial
comptime __moveinit__is_trivial = True
args_type
comptime args_type = STOutputOpArgs[dtype, layout, o]
device_type
comptime device_type = R2GOutputOp[accum_type, dtype, layout, num_threads, mma_shape, block_tile_shape, o]
Methods
__init__
__init__(out self, args: STOutputOpArgs[dtype, layout, o])
get_type_name
static get_type_name() -> String
Gets this type's name, for use in error messages when handing arguments to kernels.
Returns:
get_device_type_name
static get_device_type_name() -> String
Gets device_type's name, for use in error messages when handing arguments to kernels.
Returns:
to_kernel_args
static to_kernel_args(c: LayoutTensor[dtype, layout, o], ctx: DeviceContext) -> R2GOutputOp[accum_type, dtype, layout, num_threads, mma_shape, block_tile_shape, o].args_type
Returns:
R2GOutputOp
__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!