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: Int, n: Int, k: Int, cluster_shape: IndexList[3], num_k_partitions: Int, num_consumer: Int, partitioned_multicast: Bool, pdl_level: PDLLevel, k_group_size: Int = 0, num_pipeline_stages: Int = 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!