[all-commits] [llvm/llvm-project] 552254: [mlir][linalg] Support permutation when lowering t...

Lei Zhang via All-commits all-commits at lists.llvm.org
Mon Jan 11 06:14:03 PST 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 55225471d9838e452cfb31e0edae6162b7226221
      https://github.com/llvm/llvm-project/commit/55225471d9838e452cfb31e0edae6162b7226221
  Author: Lei Zhang <antiagainst at google.com>
  Date:   2021-01-11 (Mon, 11 Jan 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/Passes.td
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/lib/Dialect/Linalg/Transforms/Loops.cpp
    A mlir/test/Dialect/Linalg/loop-order.mlir

  Log Message:
  -----------
  [mlir][linalg] Support permutation when lowering to loop nests

Linalg ops are perfect loop nests. When materializing the concrete
loop nest, the default order specified by the Linalg op's iterators
may not be the best for further CodeGen: targets frequently need
to plan the loop order in order to gain better data access. And
different targets can have different preferences. So there should
exist a way to control the order.

Reviewed By: nicolasvasilache

Differential Revision: https://reviews.llvm.org/D91795




More information about the All-commits mailing list