Mojo function
swapAB_smallM
swapAB_smallM[a_type: DType, b_type: DType, c_type: DType, prioritize_compute_over_ctas: Bool = False, transpose_b: Bool = True](m: Scalar[DType.uint], n: Scalar[DType.uint], k: Scalar[DType.uint], cluster_shape: IndexList[3], num_k_partitions: Scalar[DType.uint], num_consumer: Scalar[DType.uint], partitioned_multicast: Bool, pdl_level: PDLLevel, k_group_size: Scalar[DType.uint] = 0, num_pipeline_stages: Scalar[DType.uint] = 0) -> MatmulConfig[a_type, b_type, c_type, transpose_b]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!