Mojo function
get_partitioned_matmul_mojo_shape
get_partitioned_matmul_mojo_shape[b_type: DType, kernel_rows: Int, kernel_cols: Int, use_i8mm: Bool](m: Int, n: Int, k: Int, num_tasks: Int) -> Index[2]
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!