Utils.cpp

Go to the documentation of this file.

//===- Utils.cpp - Transform dialect utilities ----------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#include "mlir/Dialect/Transform/Utils/Utils.h"
 
#include "mlir/IR/OpDefinition.h"
#include "mlir/Interfaces/ViewLikeInterface.h"
 
using namespace mlir;
using namespace mlir::transform;
 
void mlir::transform::printPackedOrDynamicIndexList(
    OpAsmPrinter &printer, Operation *op, Value packed, Type packedType,
    OperandRange values, TypeRange valueTypes, DenseBoolArrayAttr scalableFlags,
    DenseI64ArrayAttr integers) {
  if (packed) {
    assert(values.empty() && (!integers || integers.empty()) &&
           (!scalableFlags || scalableFlags.empty()) &&
           "expected no values/integers");
    printer << "*(" << packed;
    if (packedType) {
      printer << " : " << packedType;
    }
    printer << ")";
    return;
  }
  printDynamicIndexList(printer, op, values, integers,
                        scalableFlags ? scalableFlags.asArrayRef()
                                      : ArrayRef<bool>{},
                        valueTypes);
}
 
ParseResult mlir::transform::parsePackedOrDynamicIndexList(
    OpAsmParser &parser, std::optional<OpAsmParser::UnresolvedOperand> &packed,
    Type &packedType, SmallVectorImpl<OpAsmParser::UnresolvedOperand> &values,
    DenseBoolArrayAttr &scalableFlags, SmallVectorImpl<Type> *valueTypes,
    DenseI64ArrayAttr &integers) {
  OpAsmParser::UnresolvedOperand packedOperand;
  if (parser.parseOptionalStar().succeeded()) {
    if (parser.parseLParen().failed() ||
        parser.parseOperand(packedOperand).failed())
      return failure();
    if (packedType && (parser.parseColonType(packedType).failed()))
      return failure();
    if (parser.parseRParen().failed())
      return failure();
    packed.emplace(packedOperand);
    integers = parser.getBuilder().getDenseI64ArrayAttr({});
    scalableFlags = parser.getBuilder().getDenseBoolArrayAttr({});
    return success();
  }
 
  return parseDynamicIndexList(parser, values, integers, scalableFlags,
                               valueTypes);
}