Adam Siemieniuk 4c3db2588e [mlir][linalg] Block pack matmul pass (#89782) ...

Pack a matmul MxNxK operation into 4D blocked layout. Any present batch
dimensions remain unchanged and the result is unpacked back to the
original layout.

Matmul block packing splits the operands into major blocks (outer
dimensions) and minor blocks (inner dimensions). The desired block
layout can be controlled through packing options.

2024-05-09 18:02:44 +01:00

12 KiB

Raw Blame History

View Raw