fac349a169
…ted. (#89998)" (#90250) This partially reverts commit 7aedd7dc754c74a49fe84ed2640e269c25414087. This change removes calls to the deprecated member functions. It does not mark the functions deprecated yet and does not disable the deprecation warning in TypeSwitch. This seems to cause problems with MSVC.
136 lines
5.0 KiB
C++
136 lines
5.0 KiB
C++
//===-- TBAABuilder.cpp -- TBAA builder definitions -----------------------===//
|
|
//
|
|
// 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/CodeGen/TBAABuilder.h"
|
|
#include "flang/Optimizer/Dialect/FIRType.h"
|
|
#include "llvm/ADT/TypeSwitch.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include <mlir/Dialect/LLVMIR/LLVMAttrs.h>
|
|
#include <mlir/Dialect/LLVMIR/LLVMDialect.h>
|
|
#include <mlir/Dialect/LLVMIR/LLVMTypes.h>
|
|
|
|
#define DEBUG_TYPE "flang-tbaa-builder"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::LLVM;
|
|
|
|
static llvm::cl::opt<bool> disableTBAA(
|
|
"disable-tbaa",
|
|
llvm::cl::desc("disable attaching TBAA tags to memory accessing operations "
|
|
"to override default Flang behavior"),
|
|
llvm::cl::init(false));
|
|
|
|
// disabling this will play badly with the FIR TBAA pass, leading to worse
|
|
// performance
|
|
static llvm::cl::opt<bool> perFunctionTBAATrees(
|
|
"per-function-tbaa-trees",
|
|
llvm::cl::desc("Give each function an independent TBAA tree (default)"),
|
|
llvm::cl::init(true), llvm::cl::Hidden);
|
|
|
|
// tagAttachmentLimit is a debugging option that allows limiting
|
|
// the number of TBAA access tag attributes attached to operations.
|
|
// It is set to kTagAttachmentUnlimited by default denoting "no limit".
|
|
static constexpr unsigned kTagAttachmentUnlimited =
|
|
std::numeric_limits<unsigned>::max();
|
|
static llvm::cl::opt<unsigned>
|
|
tagAttachmentLimit("tbaa-attach-tag-max", llvm::cl::desc(""),
|
|
llvm::cl::init(kTagAttachmentUnlimited));
|
|
|
|
namespace fir {
|
|
|
|
TBAABuilder::TBAABuilder(MLIRContext *context, bool applyTBAA,
|
|
bool forceUnifiedTree)
|
|
: enableTBAA(applyTBAA && !disableTBAA),
|
|
trees(/*separatePerFunction=*/perFunctionTBAATrees && !forceUnifiedTree) {
|
|
if (!enableTBAA)
|
|
return;
|
|
}
|
|
|
|
TBAATagAttr TBAABuilder::getAccessTag(TBAATypeDescriptorAttr baseTypeDesc,
|
|
TBAATypeDescriptorAttr accessTypeDesc,
|
|
int64_t offset) {
|
|
TBAATagAttr &tag = tagsMap[{baseTypeDesc, accessTypeDesc, offset}];
|
|
if (tag)
|
|
return tag;
|
|
|
|
// Initialize new tag.
|
|
tag = TBAATagAttr::get(baseTypeDesc, accessTypeDesc, offset);
|
|
return tag;
|
|
}
|
|
|
|
TBAATagAttr TBAABuilder::getAnyBoxAccessTag(mlir::LLVM::LLVMFuncOp func) {
|
|
TBAATypeDescriptorAttr boxMemberTypeDesc = trees[func].boxMemberTypeDesc;
|
|
return getAccessTag(boxMemberTypeDesc, boxMemberTypeDesc, /*offset=*/0);
|
|
}
|
|
|
|
TBAATagAttr TBAABuilder::getBoxAccessTag(Type baseFIRType, Type accessFIRType,
|
|
GEPOp gep,
|
|
mlir::LLVM::LLVMFuncOp func) {
|
|
return getAnyBoxAccessTag(func);
|
|
}
|
|
|
|
TBAATagAttr TBAABuilder::getAnyDataAccessTag(mlir::LLVM::LLVMFuncOp func) {
|
|
TBAATypeDescriptorAttr anyDataAccessTypeDesc = trees[func].anyDataTypeDesc;
|
|
return getAccessTag(anyDataAccessTypeDesc, anyDataAccessTypeDesc,
|
|
/*offset=*/0);
|
|
}
|
|
|
|
TBAATagAttr TBAABuilder::getDataAccessTag(Type baseFIRType, Type accessFIRType,
|
|
GEPOp gep,
|
|
mlir::LLVM::LLVMFuncOp func) {
|
|
return getAnyDataAccessTag(func);
|
|
}
|
|
|
|
TBAATagAttr TBAABuilder::getAnyAccessTag(mlir::LLVM::LLVMFuncOp func) {
|
|
TBAATypeDescriptorAttr anyAccessTypeDesc = trees[func].anyAccessDesc;
|
|
return getAccessTag(anyAccessTypeDesc, anyAccessTypeDesc, /*offset=*/0);
|
|
}
|
|
|
|
void TBAABuilder::attachTBAATag(AliasAnalysisOpInterface op, Type baseFIRType,
|
|
Type accessFIRType, GEPOp gep) {
|
|
if (!enableTBAA)
|
|
return;
|
|
|
|
mlir::LLVM::LLVMFuncOp func = op->getParentOfType<mlir::LLVM::LLVMFuncOp>();
|
|
if (!func)
|
|
return;
|
|
|
|
++tagAttachmentCounter;
|
|
if (tagAttachmentLimit != kTagAttachmentUnlimited &&
|
|
tagAttachmentCounter > tagAttachmentLimit)
|
|
return;
|
|
|
|
LLVM_DEBUG(llvm::dbgs() << "Attaching TBAA tag #" << tagAttachmentCounter
|
|
<< "\n");
|
|
|
|
TBAATagAttr tbaaTagSym;
|
|
if (fir::isRecordWithDescriptorMember(baseFIRType)) {
|
|
// A memory access that addresses an aggregate that contains
|
|
// a mix of data members and descriptor members may alias
|
|
// with both data and descriptor accesses.
|
|
// Conservatively set any-access tag if there is any descriptor member.
|
|
tbaaTagSym = getAnyAccessTag(func);
|
|
} else if (mlir::isa<fir::BaseBoxType>(baseFIRType)) {
|
|
tbaaTagSym = getBoxAccessTag(baseFIRType, accessFIRType, gep, func);
|
|
} else {
|
|
tbaaTagSym = getDataAccessTag(baseFIRType, accessFIRType, gep, func);
|
|
}
|
|
|
|
if (!tbaaTagSym)
|
|
return;
|
|
|
|
op.setTBAATags(ArrayAttr::get(op->getContext(), tbaaTagSym));
|
|
}
|
|
|
|
} // namespace fir
|