diff options
| author | Nicolas Vasilache <ntv@google.com> | 2019-12-13 13:26:00 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-13 16:57:57 -0800 |
| commit | 7923abd357b82b14d1f97c2f013f41e388f77a17 (patch) | |
| tree | ab4e4716a6131085026e6855ea5632177ff1e2b7 /mlir/lib/Conversion/VectorToLoops | |
| parent | b030e4a4ec5ef47549377cc0af71a95abcf28a98 (diff) | |
| download | bcm5719-llvm-7923abd357b82b14d1f97c2f013f41e388f77a17.tar.gz bcm5719-llvm-7923abd357b82b14d1f97c2f013f41e388f77a17.zip | |
Add a layer of EDSC for linalg.GenericOp
This will be evolved into a simple programming model for custom ops and custom layers in followup CLs.
This CL also deletes the obsolete tablegen's reference-impl.td that was using EDSCs.
PiperOrigin-RevId: 285459545
Diffstat (limited to 'mlir/lib/Conversion/VectorToLoops')
| -rw-r--r-- | mlir/lib/Conversion/VectorToLoops/ConvertVectorToLoops.cpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/mlir/lib/Conversion/VectorToLoops/ConvertVectorToLoops.cpp b/mlir/lib/Conversion/VectorToLoops/ConvertVectorToLoops.cpp index 681e6e964f2..d4f362d685d 100644 --- a/mlir/lib/Conversion/VectorToLoops/ConvertVectorToLoops.cpp +++ b/mlir/lib/Conversion/VectorToLoops/ConvertVectorToLoops.cpp @@ -21,8 +21,8 @@ #include <type_traits> -#include "mlir/Dialect/VectorOps/VectorOps.h" #include "mlir/Conversion/VectorToLoops/ConvertVectorToLoops.h" +#include "mlir/Dialect/VectorOps/VectorOps.h" #include "mlir/EDSC/Builders.h" #include "mlir/EDSC/Helpers.h" #include "mlir/IR/AffineExpr.h" @@ -270,7 +270,7 @@ PatternMatchResult VectorTransferRewriter<TransferReadOp>::matchAndRewrite( VectorView vectorView(transfer.vector()); SmallVector<IndexHandle, 8> ivs = makeIndexHandles(vectorView.rank()); SmallVector<ValueHandle *, 8> pivs = - makeIndexHandlePointers(MutableArrayRef<IndexHandle>(ivs)); + makeHandlePointers(MutableArrayRef<IndexHandle>(ivs)); coalesceCopy(transfer, &pivs, &vectorView); auto lbs = vectorView.getLbs(); @@ -332,7 +332,8 @@ PatternMatchResult VectorTransferRewriter<TransferWriteOp>::matchAndRewrite( ValueHandle vectorValue(transfer.vector()); VectorView vectorView(transfer.vector()); SmallVector<IndexHandle, 8> ivs = makeIndexHandles(vectorView.rank()); - SmallVector<ValueHandle *, 8> pivs = makeIndexHandlePointers(ivs); + SmallVector<ValueHandle *, 8> pivs = + makeHandlePointers(MutableArrayRef<IndexHandle>(ivs)); coalesceCopy(transfer, &pivs, &vectorView); auto lbs = vectorView.getLbs(); |
