diff options
Diffstat (limited to 'mlir/lib/Dialect/Affine/IR/AffineOps.cpp')
| -rw-r--r-- | mlir/lib/Dialect/Affine/IR/AffineOps.cpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/mlir/lib/Dialect/Affine/IR/AffineOps.cpp b/mlir/lib/Dialect/Affine/IR/AffineOps.cpp index a7fc7dd..7f2f3c3 100644 --- a/mlir/lib/Dialect/Affine/IR/AffineOps.cpp +++ b/mlir/lib/Dialect/Affine/IR/AffineOps.cpp @@ -4474,6 +4474,17 @@ LogicalResult AffineVectorStoreOp::verify() { // DelinearizeIndexOp //===----------------------------------------------------------------------===// +LogicalResult AffineDelinearizeIndexOp::inferReturnTypes( + MLIRContext *context, std::optional<::mlir::Location> location, + ValueRange operands, DictionaryAttr attributes, OpaqueProperties properties, + RegionRange regions, SmallVectorImpl<Type> &inferredReturnTypes) { + AffineDelinearizeIndexOpAdaptor adaptor(operands, attributes, properties, + regions); + inferredReturnTypes.assign(adaptor.getBasis().size(), + IndexType::get(context)); + return success(); +} + void AffineDelinearizeIndexOp::build(OpBuilder &builder, OperationState &result, Value linearIndex, ArrayRef<OpFoldResult> basis) { |