[PATCH] D75722: [mlir][Linalg] Implement padding for linalg.conv and lowering to loops.
Han-Chung Wang via Phabricator via llvm-commits
llvm-commits at lists.llvm.org
Fri Mar 6 12:09:02 PST 2020
hanchung added inline comments.
================
Comment at: mlir/lib/Dialect/Linalg/Transforms/LinalgToLoops.cpp:178
public:
+ static ValueHandle getConvOpInput(ConvOp convOp, IndexedValueType I,
+ ArrayRef<ValueHandle> imIdx) {
----------------
mravishankar wrote:
> Could you add some comments about whats happening here?
Done.
================
Comment at: mlir/lib/Dialect/Linalg/Transforms/LinalgToLoops.cpp:206
+
+ assert(shapes[idx] != -1 && "dynamic shape is not available");
+ ValueHandle shape = std_constant_index(shapes[idx]);
----------------
mravishankar wrote:
> I am not sure we need to restrict this for the static shape case. You can always get the dimension dynamically and generate code for that. Just generate a dim operation?
Done.
Yes, you're right, we can always get the dimension here. Also change the tests to dynamic shape.
Repository:
rG LLVM Github Monorepo
CHANGES SINCE LAST ACTION
https://reviews.llvm.org/D75722/new/
https://reviews.llvm.org/D75722
More information about the llvm-commits
mailing list