1 | //===- AdaptorTest.cpp - Adaptor unit tests -------------------------------===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | |
9 | #include "../../test/lib/Dialect/Test/TestDialect.h" |
10 | #include "../../test/lib/Dialect/Test/TestOpsSyntax.h" |
11 | #include "gmock/gmock.h" |
12 | #include "gtest/gtest.h" |
13 | |
14 | using namespace llvm; |
15 | using namespace mlir; |
16 | using namespace test; |
17 | |
18 | using testing::ElementsAre; |
19 | |
20 | TEST(Adaptor, GenericAdaptorsOperandAccess) { |
21 | MLIRContext context; |
22 | context.loadDialect<test::TestDialect>(); |
23 | Builder builder(&context); |
24 | |
25 | // Has normal and Variadic arguments. |
26 | MixedNormalVariadicOperandOp::FoldAdaptor a({}); |
27 | { |
28 | SmallVector<int> v = {0, 1, 2, 3, 4}; |
29 | MixedNormalVariadicOperandOp::GenericAdaptor<ArrayRef<int>> b(v); |
30 | EXPECT_THAT(b.getInput1(), ElementsAre(0, 1)); |
31 | EXPECT_EQ(b.getInput2(), 2); |
32 | EXPECT_THAT(b.getInput3(), ElementsAre(3, 4)); |
33 | } |
34 | |
35 | // Has optional arguments. |
36 | OIListSimple::FoldAdaptor c({}, nullptr); |
37 | { |
38 | // Optional arguments return the default constructed value if not present. |
39 | // Using optional instead of plain int here to differentiate absence of |
40 | // value from the value 0. |
41 | SmallVector<std::optional<int>> v = {0, 4}; |
42 | OIListSimple::Properties prop; |
43 | prop.operandSegmentSizes = {1, 0, 1}; |
44 | OIListSimple::GenericAdaptor<ArrayRef<std::optional<int>>> d(v, {}, prop, |
45 | {}); |
46 | EXPECT_EQ(d.getArg0(), 0); |
47 | EXPECT_EQ(d.getArg1(), std::nullopt); |
48 | EXPECT_EQ(d.getArg2(), 4); |
49 | |
50 | // Check the property comparison operator. |
51 | OIListSimple::Properties equivalentProp = {1, 0, 1}; |
52 | OIListSimple::Properties differentProp = {0, 0, 1}; |
53 | EXPECT_EQ(d.getProperties(), equivalentProp); |
54 | EXPECT_NE(d.getProperties(), differentProp); |
55 | } |
56 | |
57 | // Has VariadicOfVariadic arguments. |
58 | FormatVariadicOfVariadicOperand::FoldAdaptor e({}); |
59 | { |
60 | SmallVector<int> v = {0, 1, 2, 3, 4}; |
61 | FormatVariadicOfVariadicOperand::Properties prop; |
62 | prop.operand_segments = builder.getDenseI32ArrayAttr({3, 2, 0}); |
63 | FormatVariadicOfVariadicOperand::GenericAdaptor<ArrayRef<int>> f(v, {}, |
64 | prop, {}); |
65 | SmallVector<ArrayRef<int>> operand = f.getOperand(); |
66 | ASSERT_EQ(operand.size(), (std::size_t)3); |
67 | EXPECT_THAT(operand[0], ElementsAre(0, 1, 2)); |
68 | EXPECT_THAT(operand[1], ElementsAre(3, 4)); |
69 | EXPECT_THAT(operand[2], ElementsAre()); |
70 | } |
71 | } |
72 | |