TOSA's scatter lowering is implemented by reducing accross the batch dimension.
It would be better to perform a write to the input per-input value however this
is not supported by our existing dialects. Includes a simple lit validation
test.
Details
Details
- Reviewers
sjarus NatashaKnk nicolasvasilache
Diff Detail
Diff Detail
- Repository
- rG LLVM Github Monorepo
Unit Tests
Unit Tests
Event Timeline
mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp | ||
---|---|---|
2501 | You should be able to compress those significantly: using MapList = ArrayRef<ArrayRef<AffineExpr>>; auto infer = [](MapList m) { return AffineMap::inferFromExprList(m); }; AffineExpr m, n, o, p; bindDims(rewriter.getContext(), m, n, o, p); SmallVector<AffineMap, 4> affineMaps = infer({m, n, o}, {m, p}, {m, p, o}); |
Comment Actions
Code looks good to me, I am not sure about the semantics of the op itself so someone else please cehck this.
You should be able to compress those significantly: