199d3d77dd
As proposed in https://discourse.llvm.org/t/rfc-flang-representation-for-objects-inside-physical-storage/88026, this patch adds a `storage` Value operand and a `storage_offset` Integer attribute for `[hl]fir.declare` operations. The `storage` operand indicates the raw address of the physical storage a variable belongs to. This is the beginning address of the physical storage. The `storage_offset` specifies a byte offset within the physical storage where the variable object starts.
99 lines
4.4 KiB
C++
99 lines
4.4 KiB
C++
//===-- FortranVariableInterface.cpp.cpp ----------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Coding style: https://mlir.llvm.org/getting_started/DeveloperGuide/
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "flang/Optimizer/Dialect/FortranVariableInterface.h"
|
|
|
|
#include "flang/Optimizer/Dialect/FortranVariableInterface.cpp.inc"
|
|
|
|
llvm::LogicalResult
|
|
fir::FortranVariableOpInterface::verifyDeclareLikeOpImpl(mlir::Value memref) {
|
|
const unsigned numExplicitTypeParams = getExplicitTypeParams().size();
|
|
mlir::Type memType = memref.getType();
|
|
const bool sourceIsBoxValue = mlir::isa<fir::BaseBoxType>(memType);
|
|
const bool sourceIsBoxAddress = fir::isBoxAddress(memType);
|
|
const bool sourceIsBox = sourceIsBoxValue || sourceIsBoxAddress;
|
|
if (isCharacter()) {
|
|
if (numExplicitTypeParams > 1)
|
|
return emitOpError(
|
|
"of character entity must have at most one length parameter");
|
|
if (numExplicitTypeParams == 0 && !sourceIsBox)
|
|
return emitOpError("must be provided exactly one type parameter when its "
|
|
"base is a character that is not a box");
|
|
|
|
} else if (auto recordType =
|
|
mlir::dyn_cast<fir::RecordType>(getElementType())) {
|
|
if (numExplicitTypeParams < recordType.getNumLenParams() && !sourceIsBox)
|
|
return emitOpError("must be provided all the derived type length "
|
|
"parameters when the base is not a box");
|
|
if (numExplicitTypeParams > recordType.getNumLenParams())
|
|
return emitOpError("has too many length parameters");
|
|
} else if (numExplicitTypeParams != 0) {
|
|
return emitOpError("of numeric, logical, or assumed type entity must not "
|
|
"have length parameters");
|
|
}
|
|
|
|
if (isArray()) {
|
|
if (mlir::Value shape = getShape()) {
|
|
if (sourceIsBoxAddress)
|
|
return emitOpError("for box address must not have a shape operand");
|
|
unsigned shapeRank = 0;
|
|
if (auto shapeType = mlir::dyn_cast<fir::ShapeType>(shape.getType())) {
|
|
shapeRank = shapeType.getRank();
|
|
} else if (auto shapeShiftType =
|
|
mlir::dyn_cast<fir::ShapeShiftType>(shape.getType())) {
|
|
shapeRank = shapeShiftType.getRank();
|
|
} else {
|
|
if (!sourceIsBoxValue)
|
|
emitOpError("of array entity with a raw address base must have a "
|
|
"shape operand that is a shape or shapeshift");
|
|
shapeRank = mlir::cast<fir::ShiftType>(shape.getType()).getRank();
|
|
}
|
|
|
|
std::optional<unsigned> rank = getRank();
|
|
if (!rank || *rank != shapeRank)
|
|
return emitOpError("has conflicting shape and base operand ranks");
|
|
} else if (!sourceIsBox) {
|
|
emitOpError("of array entity with a raw address base must have a shape "
|
|
"operand that is a shape or shapeshift");
|
|
}
|
|
}
|
|
return mlir::success();
|
|
}
|
|
|
|
mlir::LogicalResult
|
|
fir::detail::verifyFortranVariableStorageOpInterface(mlir::Operation *op) {
|
|
auto storageIface = mlir::cast<fir::FortranVariableStorageOpInterface>(op);
|
|
mlir::Value storage = storageIface.getStorage();
|
|
std::uint64_t storageOffset = storageIface.getStorageOffset();
|
|
if (!storage) {
|
|
if (storageOffset != 0)
|
|
return op->emitOpError(
|
|
"storage offset specified without the storage reference");
|
|
return mlir::success();
|
|
}
|
|
|
|
auto storageType =
|
|
mlir::dyn_cast<fir::SequenceType>(fir::unwrapRefType(storage.getType()));
|
|
if (!storageType || storageType.getDimension() != 1)
|
|
return op->emitOpError("storage must be a vector");
|
|
if (storageType.hasDynamicExtents())
|
|
return op->emitOpError("storage must have known extent");
|
|
if (storageType.getEleTy() != mlir::IntegerType::get(op->getContext(), 8))
|
|
return op->emitOpError("storage must be an array of i8 elements");
|
|
if (storageOffset > storageType.getConstantArraySize())
|
|
return op->emitOpError("storage offset exceeds the storage size");
|
|
// TODO: we should probably verify that the (offset + sizeof(var))
|
|
// is within the storage object, but this requires mlir::DataLayout.
|
|
// Can we make it available during the verification?
|
|
return mlir::success();
|
|
}
|