c42dd5dbb0
These interfaces allow for a composite attribute or type to opaquely provide access to any held attributes or types. There are several intended use cases for this interface. The first of which is to allow the printer to create aliases for non-builtin dialect attributes and types. In the future, this interface will also be extended to allow for SymbolRefAttr to be placed on other entities aside from just DictionaryAttr and ArrayAttr. To limit potential test breakages, this revision only adds the new interfaces to the builtin attributes/types that are currently hardcoded during AsmPrinter alias generation. In a followup the remaining builtin attributes/types, and non-builtin attributes/types can be extended to support it. Differential Revision: https://reviews.llvm.org/D102945
66 lines
2.4 KiB
C++
66 lines
2.4 KiB
C++
//===- SubElementInterfaces.cpp - Attr and Type SubElement Interfaces -----===//
|
|
//
|
|
// 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/IR/SubElementInterfaces.h"
|
|
|
|
using namespace mlir;
|
|
|
|
template <typename InterfaceT>
|
|
static void walkSubElementsImpl(InterfaceT interface,
|
|
function_ref<void(Attribute)> walkAttrsFn,
|
|
function_ref<void(Type)> walkTypesFn) {
|
|
interface.walkImmediateSubElements(
|
|
[&](Attribute attr) {
|
|
// Guard against potentially null inputs. This removes the need for the
|
|
// derived attribute/type to do it.
|
|
if (!attr)
|
|
return;
|
|
|
|
// Walk any sub elements first.
|
|
if (auto interface = attr.dyn_cast<SubElementAttrInterface>())
|
|
walkSubElementsImpl(interface, walkAttrsFn, walkTypesFn);
|
|
|
|
// Walk this attribute.
|
|
walkAttrsFn(attr);
|
|
},
|
|
[&](Type type) {
|
|
// Guard against potentially null inputs. This removes the need for the
|
|
// derived attribute/type to do it.
|
|
if (!type)
|
|
return;
|
|
|
|
// Walk any sub elements first.
|
|
if (auto interface = type.dyn_cast<SubElementTypeInterface>())
|
|
walkSubElementsImpl(interface, walkAttrsFn, walkTypesFn);
|
|
|
|
// Walk this type.
|
|
walkTypesFn(type);
|
|
});
|
|
}
|
|
|
|
void SubElementAttrInterface::walkSubElements(
|
|
function_ref<void(Attribute)> walkAttrsFn,
|
|
function_ref<void(Type)> walkTypesFn) {
|
|
assert(walkAttrsFn && walkTypesFn && "expected valid walk functions");
|
|
walkSubElementsImpl(*this, walkAttrsFn, walkTypesFn);
|
|
}
|
|
|
|
void SubElementTypeInterface::walkSubElements(
|
|
function_ref<void(Attribute)> walkAttrsFn,
|
|
function_ref<void(Type)> walkTypesFn) {
|
|
assert(walkAttrsFn && walkTypesFn && "expected valid walk functions");
|
|
walkSubElementsImpl(*this, walkAttrsFn, walkTypesFn);
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// SubElementInterface Tablegen definitions
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/IR/SubElementAttrInterfaces.cpp.inc"
|
|
#include "mlir/IR/SubElementTypeInterfaces.cpp.inc"
|