[PATCH] D73796: [mlir][Linalg] Adding support for linalg_matmul with tensors.
Lei Zhang via Phabricator via llvm-commits
llvm-commits at lists.llvm.org
Fri Jan 31 12:08:37 PST 2020
antiagainst accepted this revision.
antiagainst added a comment.
This revision is now accepted and ready to land.
Neat!
================
Comment at: mlir/test/EDSC/builder-api-test.cpp:1032
+ auto f32Type = FloatType::getF32(&globalContext());
+ auto memrefType = MemRefType::get({-1, -1}, f32Type, {}, 0);
+ auto tensorType = RankedTensorType::get({-1, -1}, f32Type);
----------------
Do we want to use `ShapedType::kDynamicSize` for these magic numbers?
================
Comment at: mlir/test/EDSC/builder-api-test.cpp:1042
+ StructuredIndexed SA(A), SB(B), SC(tensorType);
+ linalg_pointwise_add(SA({i, j}), SB({i, j}), SC({i, j}));
+ linalg_pointwise_max(SA({i, j}), SB({i, j}), SC({i, j}));
----------------
I'm not sure about the conventions here but I'd typically prefer to have small and focused tests so it's easier to maintain and upate.
Repository:
rG LLVM Github Monorepo
CHANGES SINCE LAST ACTION
https://reviews.llvm.org/D73796/new/
https://reviews.llvm.org/D73796
More information about the llvm-commits
mailing list