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
|
||||
// ABI. Once we layer that out into our own runtime types, we can
|
||||
// remove this.
|
||||
return op.getOperand().getDefiningOp<tcp::AllocMemRefOp>();
|
||||
return !op.getOperand().getDefiningOp<tcp::AllocMemRefOp>();
|
||||
});
|
||||
target.addLegalOp<tcp::GetExtentOp>();
|
||||
if (failed(applyPartialConversion(func, target, patterns))) {
|
||||
|
|
Loading…
Reference in New Issue