mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-17 14:16:42 +00:00
f326036767
I want to be able to check if the storage is contiguous in the innermost dimension, so I decided to add an entry point that takes `dim` as the number of leading dimensions to check. It seems that a runtime call might result in less code size even when `dim` is 1, so here it is. For opt-for-speed I am going to inline it in FIR. Depends on #131047.
54 lines
1.7 KiB
C++
54 lines
1.7 KiB
C++
//===-- lib/runtime/support.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/support.h"
|
|
#include "ISO_Fortran_util.h"
|
|
#include "flang-rt/runtime/descriptor.h"
|
|
#include "flang-rt/runtime/type-info.h"
|
|
|
|
namespace Fortran::runtime {
|
|
extern "C" {
|
|
RT_EXT_API_GROUP_BEGIN
|
|
|
|
bool RTDEF(IsContiguous)(const Descriptor &descriptor) {
|
|
return descriptor.IsContiguous();
|
|
}
|
|
|
|
bool RTDEF(IsContiguousUpTo)(const Descriptor &descriptor, int dim) {
|
|
return descriptor.IsContiguous(dim);
|
|
}
|
|
|
|
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) {
|
|
to = from;
|
|
if (newDynamicType) {
|
|
DescriptorAddendum *toAddendum{to.Addendum()};
|
|
INTERNAL_CHECK(toAddendum);
|
|
toAddendum->set_derivedType(newDynamicType);
|
|
to.raw().elem_len = newDynamicType->sizeInBytes();
|
|
}
|
|
to.raw().attribute = newAttribute;
|
|
if (newLowerBounds != LowerBoundModifier::Preserve) {
|
|
const ISO::CFI_index_t newLowerBound{
|
|
newLowerBounds == LowerBoundModifier::SetToOnes ? 1 : 0};
|
|
const int rank{to.rank()};
|
|
for (int i = 0; i < rank; ++i) {
|
|
to.GetDimension(i).SetLowerBound(newLowerBound);
|
|
}
|
|
}
|
|
}
|
|
|
|
RT_EXT_API_GROUP_END
|
|
} // extern "C"
|
|
} // namespace Fortran::runtime
|