Mojo function
register_epilogue
register_epilogue[MMA_M: UInt, data_paths: UInt, num_stages: UInt, bits: UInt, stage: UInt, stageN: UInt, compute_lambda_fn: fn[dtype: DType, width: Int, *, alignment: Int = 1](IndexList[2], SIMD[dtype, width]) capturing -> SIMD[dtype, width], num_output_warps: UInt, accum_type: DType, frag_size: UInt, repeats: UInt](mut upper_frag: SIMD[accum_type, frag_size], mut lower_frag: SIMD[accum_type, frag_size], c_row: UInt32, c_col: UInt32, N: UInt32)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!