This is an archive of the discontinued LLVM Phabricator instance.

Differential D93824

[DO NOT REVIEW]
AbandonedPublic

Authored by mravishankar on Dec 26 2020, 10:29 AM.

Details

Reviewers
nicolasvasilache
Summary

Adapt to linalg-on-tensors changes.

Diff Detail

Revision Contents

PathSize
mlir/
lib/
Dialect/
Linalg/
Transforms/
4 lines

Diff 315439

mlir/lib/Dialect/Linalg/Transforms/Sparsification.cpp

Show First 20 Lines • Show All 460 Lines • ▼ Show 20 Lines
static unsigned buildLattices(Merger &merger, linalg::GenericOp op, static unsigned buildLattices(Merger &merger, linalg::GenericOp op,
unsigned exp, unsigned idx) { unsigned exp, unsigned idx) {
Kind kind = merger.exp(exp).kind; Kind kind = merger.exp(exp).kind;
if (kind == Kind::kTensor || kind == Kind::kInvariant) { if (kind == Kind::kTensor || kind == Kind::kInvariant) {
// Either the index is really used in the tensor expression, or it is // Either the index is really used in the tensor expression, or it is
// set to the undefined index in that dimension. An invariant expression // set to the undefined index in that dimension. An invariant expression
// is set to a synthetic tensor with undefined indices only. // is set to a synthetic tensor with undefined indices only.
unsigned s = merger.addSet(); unsigned s = merger.addSet();
unsigned t = unsigned t = kind == Kind::kTensor ? merger.exp(exp).e0
kind == Kind::kTensor ? merger.exp(exp).e0 : op.getNumShapedOperands(); : op.getNumShapedOperands() - 1;
merger.set(s).push_back(merger.addLat(t, idx, exp)); merger.set(s).push_back(merger.addLat(t, idx, exp));
return s; return s;
} }
unsigned s0 = buildLattices(merger, op, merger.exp(exp).e0, idx); unsigned s0 = buildLattices(merger, op, merger.exp(exp).e0, idx);
unsigned s1 = buildLattices(merger, op, merger.exp(exp).e1, idx); unsigned s1 = buildLattices(merger, op, merger.exp(exp).e1, idx);
switch (kind) { switch (kind) {
case Kind::kTensor: case Kind::kTensor:
case Kind::kInvariant: case Kind::kInvariant:
▲ Show 20 Lines • Show All 658 Lines • Show Last 20 Lines