[all-commits] [llvm/llvm-project] 4c3db2: [mlir][linalg] Block pack matmul pass (#89782)

Adam Siemieniuk via All-commits all-commits at lists.llvm.org
Thu May 9 10:03:05 PDT 2024


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 4c3db2588e8b38f75744def6e2dd17c556950e46
      https://github.com/llvm/llvm-project/commit/4c3db2588e8b38f75744def6e2dd17c556950e46
  Author: Adam Siemieniuk <adam.siemieniuk at intel.com>
  Date:   2024-05-09 (Thu, 09 May 2024)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/Passes.td
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    A mlir/lib/Dialect/Linalg/Transforms/BlockPackMatmul.cpp
    M mlir/lib/Dialect/Linalg/Transforms/CMakeLists.txt
    A mlir/test/Dialect/Linalg/block-pack-matmul-layout.mlir
    A mlir/test/Dialect/Linalg/block-pack-matmul-padding.mlir
    A mlir/test/Dialect/Linalg/block-pack-matmul.mlir

  Log Message:
  -----------
  [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.



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications


More information about the All-commits mailing list