yggdrasil/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[flang][runtime] add IsAssumedSize API (#93857)

Browse Source 
Needed for SELECT RANK implementation. I want to stay away from
generating the `rank > 0 && ...` logic in FIR codegen for now.
This commit is contained in:
jeanPerier 2024-05-31 08:37:23 +02:00 committed by GitHub
parent 4985f25ffc
commit f49d26bc77
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
flang/include/flang/Runtime/support.h
View File

@ -34,6 +34,9 @@ extern "C" {
// Predicate: is the storage described by a Descriptor contiguous in memory?
bool RTDECL(IsContiguous)(const Descriptor &);
// Predicate: is this descriptor describing an assumed-size array?
bool RTDECL(IsAssumedSize)(const Descriptor &);
// Copy "from" descriptor into "to" descriptor and update "to" dynamic type,
// CFI_attribute, and lower bounds according to the other arguments.
// "newDynamicType" may be a null pointer in which case "to" dynamic type is the

5
flang/runtime/support.cpp
View File

@ -7,6 +7,7 @@
//===----------------------------------------------------------------------===//
#include "flang/Runtime/support.h"
#include "ISO_Fortran_util.h"
#include "type-info.h"
#include "flang/Runtime/descriptor.h"
@ -18,6 +19,10 @@ bool RTDEF(IsContiguous)(const Descriptor &descriptor) {
return descriptor.IsContiguous();
}
bool RTDEF(IsAssumedSize)(const Descriptor &descriptor) {
return ISO::IsAssumedSize(&descriptor.raw());
}
void RTDEF(CopyAndUpdateDescriptor)(Descriptor &to, const Descriptor &from,
const typeInfo::DerivedType *newDynamicType,
ISO::CFI_attribute_t newAttribute, enum LowerBoundModifier newLowerBounds) {

11
flang/unittests/Runtime/Support.cpp
View File

@ -56,3 +56,14 @@ TEST(CopyAndUpdateDescriptor, Basic) {
EXPECT_EQ(result.GetDimension(1).Extent(), x->GetDimension(1).Extent());
EXPECT_EQ(result.GetDimension(1).LowerBound(), 1);
}
TEST(IsAssumedSize, Basic) {
auto x{MakeArray<TypeCategory::Integer, 4>(
std::vector<int>{2, 3}, std::vector<std::int32_t>{0, 1, 2, 3, 4, 5})};
EXPECT_FALSE(RTNAME(IsAssumedSize)(*x));
x->GetDimension(1).SetExtent(-1);
EXPECT_TRUE(RTNAME(IsAssumedSize)(*x));
auto scalar{MakeArray<TypeCategory::Integer, 4>(
std::vector<int>{}, std::vector<std::int32_t>{0})};
EXPECT_FALSE(RTNAME(IsAssumedSize)(*scalar));
}