201 lines
8.7 KiB
C++
201 lines
8.7 KiB
C++
|
//===-- 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)));
|
||
|
|
}
|