forked from OSchip/llvm-project
82 lines
3.0 KiB
C++
82 lines
3.0 KiB
C++
//===- PDLInterp.cpp - PDL Interpreter Dialect ------------------*- C++ -*-===//
|
|
//
|
|
// 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/PDLInterp/IR/PDLInterp.h"
|
|
#include "mlir/Dialect/PDL/IR/PDLTypes.h"
|
|
#include "mlir/IR/BuiltinTypes.h"
|
|
#include "mlir/IR/DialectImplementation.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::pdl_interp;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// PDLInterp Dialect
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
void PDLInterpDialect::initialize() {
|
|
addOperations<
|
|
#define GET_OP_LIST
|
|
#include "mlir/Dialect/PDLInterp/IR/PDLInterpOps.cpp.inc"
|
|
>();
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// pdl_interp::CreateOperationOp
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
static ParseResult parseCreateOperationOpAttributes(
|
|
OpAsmParser &p, SmallVectorImpl<OpAsmParser::OperandType> &attrOperands,
|
|
ArrayAttr &attrNamesAttr) {
|
|
Builder &builder = p.getBuilder();
|
|
SmallVector<Attribute, 4> attrNames;
|
|
if (succeeded(p.parseOptionalLBrace())) {
|
|
do {
|
|
StringAttr nameAttr;
|
|
OpAsmParser::OperandType operand;
|
|
if (p.parseAttribute(nameAttr) || p.parseEqual() ||
|
|
p.parseOperand(operand))
|
|
return failure();
|
|
attrNames.push_back(nameAttr);
|
|
attrOperands.push_back(operand);
|
|
} while (succeeded(p.parseOptionalComma()));
|
|
if (p.parseRBrace())
|
|
return failure();
|
|
}
|
|
attrNamesAttr = builder.getArrayAttr(attrNames);
|
|
return success();
|
|
}
|
|
|
|
static void printCreateOperationOpAttributes(OpAsmPrinter &p,
|
|
CreateOperationOp op,
|
|
OperandRange attrArgs,
|
|
ArrayAttr attrNames) {
|
|
if (attrNames.empty())
|
|
return;
|
|
p << " {";
|
|
interleaveComma(llvm::seq<int>(0, attrNames.size()), p,
|
|
[&](int i) { p << attrNames[i] << " = " << attrArgs[i]; });
|
|
p << '}';
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// pdl_interp::GetValueTypeOp
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
/// Given the result type of a `GetValueTypeOp`, return the expected input type.
|
|
static Type getGetValueTypeOpValueType(Type type) {
|
|
Type valueTy = pdl::ValueType::get(type.getContext());
|
|
return type.isa<pdl::RangeType>() ? pdl::RangeType::get(valueTy) : valueTy;
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// TableGen Auto-Generated Op and Interface Definitions
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#define GET_OP_CLASSES
|
|
#include "mlir/Dialect/PDLInterp/IR/PDLInterpOps.cpp.inc"
|