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,
Copyable,
DevicePassable,
ImplicitlyCopyable,
OutputOp,
UnknownDestructibility
Aliases
__copyinit__is_trivial
alias __copyinit__is_trivial = True
__del__is_trivial
alias __del__is_trivial = True
args_type
alias args_type = STOutputOpArgs[dtype, layout, o]
device_type
alias 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) -> 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!