[all-commits] [llvm/llvm-project] 7e7182: [mlir][linalg] Restrict distribution to parallel dims

[Lei Zhang via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 7e71823f1deb54a1465bc4040f4e3158357f71df
      https://github.com/llvm/llvm-project/commit/7e71823f1deb54a1465bc4040f4e3158357f71df
  Author: Lei Zhang <antiagainst at google.com>
  Date:   2021-05-10 (Mon, 10 May 2021)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Utils/Utils.cpp
    M mlir/test/Dialect/Linalg/tile-and-distribute.mlir
    M mlir/test/lib/Transforms/TestLinalgTransforms.cpp

  Log Message:
  -----------
  [mlir][linalg] Restrict distribution to parallel dims

According to the API contract, LinalgLoopDistributionOptions
expects to work on parallel iterators. When getting processor
information, only loop ranges for parallel dimensions should
be fed in. But right now after generating scf.for loop nests,
we feed in *all* loops, including the ones materialized for
reduction iterators. This can cause unexpected distribution
of reduction dimensions. This commit fixes it.

Reviewed By: mravishankar

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