[all-commits] [llvm/llvm-project] f365e8: [mlir] Revisit `LinalgLoopDistributionOptions`.

MaheshRavishankar via All-commits all-commits at lists.llvm.org
Mon Aug 15 08:56:47 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f365e85c8314657e0c0f9257fbcb367d74e9d0cd
      https://github.com/llvm/llvm-project/commit/f365e85c8314657e0c0f9257fbcb367d74e9d0cd
  Author: Mahesh Ravishankar <ravishankarm at google.com>
  Date:   2022-08-15 (Mon, 15 Aug 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/Utils/Utils.h
    M mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
    M mlir/lib/Dialect/Linalg/Utils/Utils.cpp
    M mlir/test/lib/Dialect/Linalg/TestLinalgTransforms.cpp

  Log Message:
  -----------
  [mlir] Revisit `LinalgLoopDistributionOptions`.

This patch cleans up the way `LinalgLoopDistributionOptions` are meant
to be used. The option just contains a call back that takes the list
of loop ranges that represent the loops that are to be distributed.
These loops are the outer parallel loops of the tiled operation which
have non-zero tile sizes specified. The call back returns for each of
the loops,
- The procId to use,
- The number of processors,
- The distribution method to use for that loop.

Reviewed By: antiagainst, hanchung

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




More information about the All-commits mailing list