| //===-- flang/unittests/Runtime/Matmul.cpp--------- -------------*- 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 "flang/Runtime/matmul.h" |
| #include "gtest/gtest.h" |
| #include "tools.h" |
| #include "flang/Runtime/allocatable.h" |
| #include "flang/Runtime/cpp-type.h" |
| #include "flang/Runtime/descriptor.h" |
| #include "flang/Runtime/type-code.h" |
| |
| using namespace Fortran::runtime; |
| using Fortran::common::TypeCategory; |
| |
| TEST(Matmul, Basic) { |
| // X 0 2 4 Y 6 9 V -1 -2 |
| // 1 3 5 7 10 |
| // 8 11 |
| auto x{MakeArray<TypeCategory::Integer, 4>( |
| std::vector<int>{2, 3}, std::vector<std::int32_t>{0, 1, 2, 3, 4, 5})}; |
| auto y{MakeArray<TypeCategory::Integer, 2>( |
| std::vector<int>{3, 2}, std::vector<std::int16_t>{6, 7, 8, 9, 10, 11})}; |
| auto v{MakeArray<TypeCategory::Integer, 8>( |
| std::vector<int>{2}, std::vector<std::int64_t>{-1, -2})}; |
| |
| // X2 0 2 4 Y2 -1 -1 |
| // 1 3 5 6 9 |
| // -1 -1 -1 7 10 |
| // 8 11 |
| auto x2{MakeArray<TypeCategory::Integer, 4>(std::vector<int>{3, 3}, |
| std::vector<std::int32_t>{0, 1, -1, 2, 3, -1, 4, 5})}; |
| auto y2{MakeArray<TypeCategory::Integer, 2>(std::vector<int>{4, 2}, |
| std::vector<std::int16_t>{-1, 6, 7, 8, -1, 9, 10, 11})}; |
| |
| StaticDescriptor<2, true> statDesc; |
| Descriptor &result{statDesc.descriptor()}; |
| |
| RTNAME(Matmul)(result, *x, *y, __FILE__, __LINE__); |
| ASSERT_EQ(result.rank(), 2); |
| EXPECT_EQ(result.GetDimension(0).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(0).Extent(), 2); |
| EXPECT_EQ(result.GetDimension(1).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(1).Extent(), 2); |
| ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 4})); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(0), 46); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(1), 67); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(2), 64); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(3), 94); |
| |
| std::memset( |
| result.raw().base_addr, 0, result.Elements() * result.ElementBytes()); |
| result.GetDimension(0).SetLowerBound(0); |
| result.GetDimension(1).SetLowerBound(2); |
| RTNAME(MatmulDirect)(result, *x, *y, __FILE__, __LINE__); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(0), 46); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(1), 67); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(2), 64); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(3), 94); |
| result.Destroy(); |
| |
| RTNAME(Matmul)(result, *v, *x, __FILE__, __LINE__); |
| ASSERT_EQ(result.rank(), 1); |
| EXPECT_EQ(result.GetDimension(0).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(0).Extent(), 3); |
| ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 8})); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(0), -2); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(1), -8); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(2), -14); |
| result.Destroy(); |
| |
| RTNAME(Matmul)(result, *y, *v, __FILE__, __LINE__); |
| ASSERT_EQ(result.rank(), 1); |
| EXPECT_EQ(result.GetDimension(0).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(0).Extent(), 3); |
| ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 8})); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(0), -24); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(1), -27); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(2), -30); |
| result.Destroy(); |
| |
| // Test non-contiguous sections. |
| static constexpr int sectionRank{2}; |
| StaticDescriptor<sectionRank> sectionStaticDescriptorX2; |
| Descriptor §ionX2{sectionStaticDescriptorX2.descriptor()}; |
| sectionX2.Establish(x2->type(), x2->ElementBytes(), |
| /*p=*/nullptr, /*rank=*/sectionRank); |
| static const SubscriptValue lowersX2[]{1, 1}, uppersX2[]{2, 3}; |
| // Section of X2: |
| // +--------+ |
| // | 0 2 4| |
| // | 1 3 5| |
| // +--------+ |
| // -1 -1 -1 |
| const auto errorX2{CFI_section( |
| §ionX2.raw(), &x2->raw(), lowersX2, uppersX2, /*strides=*/nullptr)}; |
| ASSERT_EQ(errorX2, 0) << "CFI_section failed for X2: " << errorX2; |
| |
| StaticDescriptor<sectionRank> sectionStaticDescriptorY2; |
| Descriptor §ionY2{sectionStaticDescriptorY2.descriptor()}; |
| sectionY2.Establish(y2->type(), y2->ElementBytes(), |
| /*p=*/nullptr, /*rank=*/sectionRank); |
| static const SubscriptValue lowersY2[]{2, 1}; |
| // Section of Y2: |
| // -1 -1 |
| // +-----+ |
| // | 6 9| |
| // | 7 10| |
| // | 8 11| |
| // +-----+ |
| const auto errorY2{CFI_section(§ionY2.raw(), &y2->raw(), lowersY2, |
| /*uppers=*/nullptr, /*strides=*/nullptr)}; |
| ASSERT_EQ(errorY2, 0) << "CFI_section failed for Y2: " << errorY2; |
| |
| RTNAME(Matmul)(result, sectionX2, *y, __FILE__, __LINE__); |
| ASSERT_EQ(result.rank(), 2); |
| EXPECT_EQ(result.GetDimension(0).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(0).Extent(), 2); |
| EXPECT_EQ(result.GetDimension(1).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(1).Extent(), 2); |
| ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 4})); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(0), 46); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(1), 67); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(2), 64); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(3), 94); |
| result.Destroy(); |
| |
| RTNAME(Matmul)(result, *x, sectionY2, __FILE__, __LINE__); |
| ASSERT_EQ(result.rank(), 2); |
| EXPECT_EQ(result.GetDimension(0).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(0).Extent(), 2); |
| EXPECT_EQ(result.GetDimension(1).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(1).Extent(), 2); |
| ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 4})); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(0), 46); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(1), 67); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(2), 64); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(3), 94); |
| result.Destroy(); |
| |
| RTNAME(Matmul)(result, sectionX2, sectionY2, __FILE__, __LINE__); |
| ASSERT_EQ(result.rank(), 2); |
| EXPECT_EQ(result.GetDimension(0).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(0).Extent(), 2); |
| EXPECT_EQ(result.GetDimension(1).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(1).Extent(), 2); |
| ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 4})); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(0), 46); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(1), 67); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(2), 64); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int32_t>(3), 94); |
| result.Destroy(); |
| |
| RTNAME(Matmul)(result, *v, sectionX2, __FILE__, __LINE__); |
| ASSERT_EQ(result.rank(), 1); |
| EXPECT_EQ(result.GetDimension(0).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(0).Extent(), 3); |
| ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 8})); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(0), -2); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(1), -8); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(2), -14); |
| result.Destroy(); |
| |
| RTNAME(Matmul)(result, sectionY2, *v, __FILE__, __LINE__); |
| ASSERT_EQ(result.rank(), 1); |
| EXPECT_EQ(result.GetDimension(0).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(0).Extent(), 3); |
| ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Integer, 8})); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(0), -24); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(1), -27); |
| EXPECT_EQ(*result.ZeroBasedIndexedElement<std::int64_t>(2), -30); |
| result.Destroy(); |
| |
| // X F F T Y F T |
| // F T T F T |
| // F F |
| auto xLog{MakeArray<TypeCategory::Logical, 1>(std::vector<int>{2, 3}, |
| std::vector<std::uint8_t>{false, false, false, true, true, false})}; |
| auto yLog{MakeArray<TypeCategory::Logical, 2>(std::vector<int>{3, 2}, |
| std::vector<std::uint16_t>{false, false, false, true, true, false})}; |
| RTNAME(Matmul)(result, *xLog, *yLog, __FILE__, __LINE__); |
| ASSERT_EQ(result.rank(), 2); |
| EXPECT_EQ(result.GetDimension(0).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(0).Extent(), 2); |
| EXPECT_EQ(result.GetDimension(1).LowerBound(), 1); |
| EXPECT_EQ(result.GetDimension(1).Extent(), 2); |
| ASSERT_EQ(result.type(), (TypeCode{TypeCategory::Logical, 2})); |
| EXPECT_FALSE( |
| static_cast<bool>(*result.ZeroBasedIndexedElement<std::uint16_t>(0))); |
| EXPECT_FALSE( |
| static_cast<bool>(*result.ZeroBasedIndexedElement<std::uint16_t>(1))); |
| EXPECT_FALSE( |
| static_cast<bool>(*result.ZeroBasedIndexedElement<std::uint16_t>(2))); |
| EXPECT_TRUE( |
| static_cast<bool>(*result.ZeroBasedIndexedElement<std::uint16_t>(3))); |
| } |