Mojo function

pack_matmul_b_shape_func

pack_matmul_b_shape_func[a_type: DType, c_type: DType, transpose_in_0: Bool](b_input: TileTensor[linear_idx_type=b_input.linear_idx_type, element_size=b_input.element_size], kernel_type_m: Int = 0) -> IndexList[2]

TileTensor primary implementation of pack_matmul_b_shape_func.

Takes kernel_type_m directly instead of extracting it from a_shape static shape params (0 = dynamic M).

Returns:

IndexList[2]

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit