Mojo function

pack_b_ndbuffer

pack_b_ndbuffer[b_type: DType, //, a_type: DType, c_type: DType](b_input: TileTensor[b_type, b_input.LayoutType, b_input.origin, linear_idx_type=b_input.linear_idx_type, element_size=b_input.element_size], output_buffer: TileTensor[b_type, output_buffer.LayoutType, output_buffer.origin, linear_idx_type=output_buffer.linear_idx_type, element_size=output_buffer.element_size], kernel_type_m: Int = 0)

TileTensor primary implementation of pack_b_ndbuffer.

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

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