mirror of https://github.com/llvm/torch-mlir
Fix typo in LowerLinalgLoopDimOps
The legality condition was reversed. It's unclear to me why this didn't cause "failed to legalize".pull/1/head
parent
836a8d4bec
commit
9191efdfc1
|
@ -222,7 +222,7 @@ class LowerLinalgLoopDimOps
|
||||||
// TODO: We only need this because we use `dim` ops for the memref
|
// TODO: We only need this because we use `dim` ops for the memref
|
||||||
// ABI. Once we layer that out into our own runtime types, we can
|
// ABI. Once we layer that out into our own runtime types, we can
|
||||||
// remove this.
|
// remove this.
|
||||||
return op.getOperand().getDefiningOp<tcp::AllocMemRefOp>();
|
return !op.getOperand().getDefiningOp<tcp::AllocMemRefOp>();
|
||||||
});
|
});
|
||||||
target.addLegalOp<tcp::GetExtentOp>();
|
target.addLegalOp<tcp::GetExtentOp>();
|
||||||
if (failed(applyPartialConversion(func, target, patterns))) {
|
if (failed(applyPartialConversion(func, target, patterns))) {
|
||||||
|
|
Loading…
Reference in New Issue