diff options
| author | Lei Zhang <antiagainst@google.com> | 2021-05-10 15:17:14 -0400 |
|---|---|---|
| committer | Lei Zhang <antiagainst@google.com> | 2021-05-10 15:23:00 -0400 |
| commit | 7e71823f1deb54a1465bc4040f4e3158357f71df (patch) | |
| tree | b656cba610c4b2cb371265634cd9bdda33bb87b5 /llvm/lib/CodeGen/TargetLoweringObjectFileImpl.cpp | |
| parent | 0c64cef8943546fce28936bb51cce2e22f88c698 (diff) | |
| download | llvm-7e71823f1deb54a1465bc4040f4e3158357f71df.zip llvm-7e71823f1deb54a1465bc4040f4e3158357f71df.tar.gz llvm-7e71823f1deb54a1465bc4040f4e3158357f71df.tar.bz2 | |
[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
Diffstat (limited to 'llvm/lib/CodeGen/TargetLoweringObjectFileImpl.cpp')
0 files changed, 0 insertions, 0 deletions