1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
//===- SubElementInterfaceTest.cpp - SubElementInterface unit tests -------===//
//
// 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/Builders.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/IR/SubElementInterfaces.h"
#include "gtest/gtest.h"
#include <cstdint>
using namespace mlir;
using namespace mlir::detail;
namespace {
TEST(SubElementInterfaceTest, Nested) {
MLIRContext context;
Builder builder(&context);
BoolAttr trueAttr = builder.getBoolAttr(true);
BoolAttr falseAttr = builder.getBoolAttr(false);
ArrayAttr boolArrayAttr = builder.getArrayAttr({trueAttr, falseAttr});
DictionaryAttr dictAttr =
builder.getDictionaryAttr(builder.getNamedAttr("array", boolArrayAttr));
SmallVector<Attribute> subAttrs;
dictAttr.walkSubAttrs([&](Attribute attr) { subAttrs.push_back(attr); });
EXPECT_EQ(llvm::makeArrayRef(subAttrs),
ArrayRef<Attribute>({trueAttr, falseAttr, boolArrayAttr}));
}
} // end namespace
|
