[all-commits] [llvm/llvm-project] 18b92c: [mlir][Linalg] Add a TileToForeachThread transform.

Nicolas Vasilache via All-commits all-commits at lists.llvm.org
Tue Jul 19 04:59:26 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 18b92c66fe59a44f50bc211a418eaf48fe1cf7c1
      https://github.com/llvm/llvm-project/commit/18b92c66fe59a44f50bc211a418eaf48fe1cf7c1
  Author: Nicolas Vasilache <nicolas.vasilache at gmail.com>
  Date:   2022-07-19 (Tue, 19 Jul 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.h
    M mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/include/mlir/Dialect/Utils/StaticValueUtils.h
    M mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp
    M mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
    M mlir/lib/Dialect/Utils/StaticValueUtils.cpp
    A mlir/test/Dialect/Linalg/tile-to-foreach-thread.mlir

  Log Message:
  -----------
  [mlir][Linalg] Add a TileToForeachThread transform.

This revision adds a new transformation to tile a TilingInterface `op` to a tiled `scf.foreach_thread`, applying
tiling by `num_threads`.
If non-empty, the `threadDimMapping` is added as an attribute to the resulting `scf.foreach_thread`.
0-tile sizes (i.e. tile by the full size of the data) are used to encode
that a dimension is not tiled.

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




More information about the All-commits mailing list