1//===-- unittests/Runtime/Matmul.cpp ----------------------------*- C++ -*-===//
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 "flang/Runtime/matmul.h"
10#include "tools.h"
11#include "gtest/gtest.h"
12#include "flang-rt/runtime/descriptor.h"
13#include "flang-rt/runtime/type-code.h"
14#include "flang/Runtime/allocatable.h"
15#include "flang/Runtime/cpp-type.h"
16
17using namespace Fortran::runtime;
18using Fortran::common::TypeCategory;
19
20TEST(Matmul, Basic) {
21 // X 0 2 4 Y 6 9 V -1 -2
22 // 1 3 5 7 10
23 // 8 11
24 auto x{MakeArray<TypeCategory::Integer, 4>(
25 std::vector<int>{2, 3}, std::vector<std::int32_t>{0, 1, 2, 3, 4, 5})};
26 auto y{MakeArray<TypeCategory::Integer, 2>(
27 std::vector<int>{3, 2}, std::vector<std::int16_t>{6, 7, 8, 9, 10, 11})};
28 auto v{MakeArray<TypeCategory::Integer, 8>(
29 std::vector<int>{2}, std::vector<std::int64_t>{-1, -2})};
30
31 // X2 0 2 4 Y2 -1 -1
32 // 1 3 5 6 9
33 // -1 -1 -1 7 10
34 // 8 11
35 auto x2{MakeArray<TypeCategory::Integer, 4>(std::vector<int>{3, 3},
36 std::vector<std::int32_t>{0, 1, -1, 2, 3, -1, 4, 5})};
37 auto y2{MakeArray<TypeCategory::Integer, 2>(std::vector<int>{4, 2},
38 std::vector<std::int16_t>{-1, 6, 7, 8, -1, 9, 10, 11})};
39
40 StaticDescriptor<2, true> statDesc;
41 Descriptor &result{statDesc.descriptor()};
42
43 RTNAME(MatmulInteger4Integer2)(result, *x, *y, __FILE__, __LINE__);
44 ASSERT_EQ(result.rank(), 2);
45 EXPECT_EQ(result.GetDimension(0).LowerBound(), 1);
46 EXPECT_EQ(result.GetDimension(0).Extent(), 2);
47 EXPECT_EQ(result.GetDimension(1).LowerBound(), 1);
48 EXPECT_EQ(result.GetDimension(1).Extent(), 2);
49 ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 4}));
50 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(0), 46);
51 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(1), 67);
52 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(2), 64);
53 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(3), 94);
54
55 std::memset(
56 s: result.raw().base_addr, c: 0, n: result.Elements() * result.ElementBytes());
57 result.GetDimension(0).SetLowerBound(0);
58 result.GetDimension(1).SetLowerBound(2);
59 RTNAME(MatmulDirectInteger4Integer2)(result, *x, *y, __FILE__, __LINE__);
60 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(0), 46);
61 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(1), 67);
62 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(2), 64);
63 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(3), 94);
64 result.Destroy();
65
66 RTNAME(MatmulInteger8Integer4)(result, *v, *x, __FILE__, __LINE__);
67 ASSERT_EQ(result.rank(), 1);
68 EXPECT_EQ(result.GetDimension(0).LowerBound(), 1);
69 EXPECT_EQ(result.GetDimension(0).Extent(), 3);
70 ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 8}));
71 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(0), -2);
72 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(1), -8);
73 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(2), -14);
74 result.Destroy();
75
76 RTNAME(MatmulInteger2Integer8)(result, *y, *v, __FILE__, __LINE__);
77 ASSERT_EQ(result.rank(), 1);
78 EXPECT_EQ(result.GetDimension(0).LowerBound(), 1);
79 EXPECT_EQ(result.GetDimension(0).Extent(), 3);
80 ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 8}));
81 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(0), -24);
82 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(1), -27);
83 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(2), -30);
84 result.Destroy();
85
86 // Test non-contiguous sections.
87 static constexpr int sectionRank{2};
88 StaticDescriptor<sectionRank> sectionStaticDescriptorX2;
89 Descriptor &sectionX2{sectionStaticDescriptorX2.descriptor()};
90 sectionX2.Establish(x2->type(), x2->ElementBytes(),
91 /*p=*/nullptr, /*rank=*/sectionRank);
92 static const SubscriptValue lowersX2[]{1, 1}, uppersX2[]{2, 3};
93 // Section of X2:
94 // +--------+
95 // | 0 2 4|
96 // | 1 3 5|
97 // +--------+
98 // -1 -1 -1
99 const auto errorX2{CFI_section(
100 &sectionX2.raw(), &x2->raw(), lowersX2, uppersX2, /*strides=*/nullptr)};
101 ASSERT_EQ(errorX2, 0) << "CFI_section failed for X2: " << errorX2;
102
103 StaticDescriptor<sectionRank> sectionStaticDescriptorY2;
104 Descriptor &sectionY2{sectionStaticDescriptorY2.descriptor()};
105 sectionY2.Establish(y2->type(), y2->ElementBytes(),
106 /*p=*/nullptr, /*rank=*/sectionRank);
107 static const SubscriptValue lowersY2[]{2, 1};
108 // Section of Y2:
109 // -1 -1
110 // +-----+
111 // | 6 9|
112 // | 7 10|
113 // | 8 11|
114 // +-----+
115 const auto errorY2{CFI_section(&sectionY2.raw(), &y2->raw(), lowersY2,
116 /*uppers=*/nullptr, /*strides=*/nullptr)};
117 ASSERT_EQ(errorY2, 0) << "CFI_section failed for Y2: " << errorY2;
118
119 RTNAME(MatmulInteger4Integer2)(result, sectionX2, *y, __FILE__, __LINE__);
120 ASSERT_EQ(result.rank(), 2);
121 EXPECT_EQ(result.GetDimension(0).LowerBound(), 1);
122 EXPECT_EQ(result.GetDimension(0).Extent(), 2);
123 EXPECT_EQ(result.GetDimension(1).LowerBound(), 1);
124 EXPECT_EQ(result.GetDimension(1).Extent(), 2);
125 ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 4}));
126 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(0), 46);
127 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(1), 67);
128 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(2), 64);
129 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(3), 94);
130 result.Destroy();
131
132 RTNAME(MatmulInteger4Integer2)(result, *x, sectionY2, __FILE__, __LINE__);
133 ASSERT_EQ(result.rank(), 2);
134 EXPECT_EQ(result.GetDimension(0).LowerBound(), 1);
135 EXPECT_EQ(result.GetDimension(0).Extent(), 2);
136 EXPECT_EQ(result.GetDimension(1).LowerBound(), 1);
137 EXPECT_EQ(result.GetDimension(1).Extent(), 2);
138 ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 4}));
139 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(0), 46);
140 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(1), 67);
141 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(2), 64);
142 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(3), 94);
143 result.Destroy();
144
145 RTNAME(MatmulInteger4Integer2)
146 (result, sectionX2, sectionY2, __FILE__, __LINE__);
147 ASSERT_EQ(result.rank(), 2);
148 EXPECT_EQ(result.GetDimension(0).LowerBound(), 1);
149 EXPECT_EQ(result.GetDimension(0).Extent(), 2);
150 EXPECT_EQ(result.GetDimension(1).LowerBound(), 1);
151 EXPECT_EQ(result.GetDimension(1).Extent(), 2);
152 ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 4}));
153 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(0), 46);
154 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(1), 67);
155 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(2), 64);
156 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(3), 94);
157 result.Destroy();
158
159 RTNAME(MatmulInteger8Integer4)(result, *v, sectionX2, __FILE__, __LINE__);
160 ASSERT_EQ(result.rank(), 1);
161 EXPECT_EQ(result.GetDimension(0).LowerBound(), 1);
162 EXPECT_EQ(result.GetDimension(0).Extent(), 3);
163 ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 8}));
164 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(0), -2);
165 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(1), -8);
166 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(2), -14);
167 result.Destroy();
168
169 RTNAME(MatmulInteger2Integer8)(result, sectionY2, *v, __FILE__, __LINE__);
170 ASSERT_EQ(result.rank(), 1);
171 EXPECT_EQ(result.GetDimension(0).LowerBound(), 1);
172 EXPECT_EQ(result.GetDimension(0).Extent(), 3);
173 ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 8}));
174 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(0), -24);
175 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(1), -27);
176 EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(2), -30);
177 result.Destroy();
178
179 // X F F T Y F T
180 // F T T F T
181 // F F
182 auto xLog{MakeArray<TypeCategory::Logical, 1>(std::vector<int>{2, 3},
183 std::vector<std::uint8_t>{false, false, false, true, true, false})};
184 auto yLog{MakeArray<TypeCategory::Logical, 2>(std::vector<int>{3, 2},
185 std::vector<std::uint16_t>{false, false, false, true, true, false})};
186 RTNAME(MatmulLogical1Logical2)(result, *xLog, *yLog, __FILE__, __LINE__);
187 ASSERT_EQ(result.rank(), 2);
188 EXPECT_EQ(result.GetDimension(0).LowerBound(), 1);
189 EXPECT_EQ(result.GetDimension(0).Extent(), 2);
190 EXPECT_EQ(result.GetDimension(1).LowerBound(), 1);
191 EXPECT_EQ(result.GetDimension(1).Extent(), 2);
192 ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Logical, 2}));
193 EXPECT_FALSE(
194 static_cast<bool>(*result.ZeroBasedIndexedElement<std::uint16_t>(0)));
195 EXPECT_FALSE(
196 static_cast<bool>(*result.ZeroBasedIndexedElement<std::uint16_t>(1)));
197 EXPECT_FALSE(
198 static_cast<bool>(*result.ZeroBasedIndexedElement<std::uint16_t>(2)));
199 EXPECT_TRUE(
200 static_cast<bool>(*result.ZeroBasedIndexedElement<std::uint16_t>(3)));
201 result.Destroy();
202}
203

source code of flang-rt/unittests/Runtime/Matmul.cpp