Mojo function
bulk_mma_ws
bulk_mma_ws[kind: UMMAKind, a_dtype: DType, b_dtype: DType, *, a_BMN: Int, a_BK: Int, a_swizzle: TensorMapSwizzle, a_is_k_major: Bool, b_BMN: Int, b_BK: Int, b_swizzle: TensorMapSwizzle, b_is_k_major: Bool, num_k_mmas: Int, operand_size: Int, tcgen05_mma_type: String, mma_k: Int = 16](idesc: UMMAInsDescriptor[kind], a: MMASmemDescriptorPair, b: MMASmemDescriptorPair, c_tmem: UInt32, c_scale: UInt32, elect: Int32)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!