diff options
Diffstat (limited to 'mlir/lib/Dialect/Linalg/Transforms')
4 files changed, 3 insertions, 4 deletions
diff --git a/mlir/lib/Dialect/Linalg/Transforms/ConvertToDestinationStyle.cpp b/mlir/lib/Dialect/Linalg/Transforms/ConvertToDestinationStyle.cpp index 6801b68a8538..6c1087730ebb 100644 --- a/mlir/lib/Dialect/Linalg/Transforms/ConvertToDestinationStyle.cpp +++ b/mlir/lib/Dialect/Linalg/Transforms/ConvertToDestinationStyle.cpp @@ -553,7 +553,7 @@ Value linalg::bufferizeToAllocation( Value alloc = createAllocationForTensor( rewriter, op->getLoc(), operand->get(), options, memorySpace); allocs.push_back(alloc); - if (!state.findDefinitions(operand->get()).empty()) { + if (!state.findDefinitions(operand).empty()) { // Initialize buffer with a copy of the operand data. Not needed if the // tensor is uninitialized. createMemcpy(rewriter, op->getLoc(), operand->get(), alloc, options); diff --git a/mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp b/mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp index 0e651f4cee4c..fc6671ef8117 100644 --- a/mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp +++ b/mlir/lib/Dialect/Linalg/Transforms/Detensorize.cpp @@ -154,7 +154,6 @@ public: }); addSourceMaterialization(sourceMaterializationCallback); - addArgumentMaterialization(sourceMaterializationCallback); } }; diff --git a/mlir/lib/Dialect/Linalg/Transforms/EliminateEmptyTensors.cpp b/mlir/lib/Dialect/Linalg/Transforms/EliminateEmptyTensors.cpp index 4776883ed95c..b710bde87f9f 100644 --- a/mlir/lib/Dialect/Linalg/Transforms/EliminateEmptyTensors.cpp +++ b/mlir/lib/Dialect/Linalg/Transforms/EliminateEmptyTensors.cpp @@ -59,7 +59,7 @@ LogicalResult linalg::linalgOpAnchoredEmptyTensorEliminationStep( config.followEquivalentOnly = true; config.alwaysIncludeLeaves = false; SetVector<Value> emptyTensors = state.findValueInReverseUseDefChain( - in->get(), /*condition=*/ + in, /*condition=*/ [&](Value val) { return val.getDefiningOp<tensor::EmptyOp>() && val.getType() == in->get().getType(); diff --git a/mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp b/mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp index 60cf897b00de..50593b08ad74 100644 --- a/mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp +++ b/mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp @@ -1656,8 +1656,8 @@ void linalg::populateDecomposeConvolutionPatterns(RewritePatternSet &patterns, } void linalg::populateDecomposePackUnpackPatterns(RewritePatternSet &patterns) { - // TODO: Add and test patterns for tensor.unpack patterns.add<DecomposeOuterUnitDimsPackOpPattern>(patterns.getContext()); + patterns.add<DecomposeOuterUnitDimsUnPackOpPattern>(patterns.getContext()); } void linalg::populateDecomposePadPatterns(RewritePatternSet &patterns) { |