mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-25 12:12:47 +01:00
d21f701c76
- Distinct metadata needs generating in the codegen to attach correct AAInfo on the loads/stores after lowering, merging, and other relevant transformations. - This patch adds 'MachhineModuleSlotTracker' to help assign slot numbers to these newly generated unnamed metadata nodes. - To help 'MachhineModuleSlotTracker' track machine metadata, the original 'SlotTracker' is rebased from 'AbstractSlotTrackerStorage', which provides basic interfaces to create/retrive metadata slots. In addition, once LLVM IR is processsed, additional hooks are also introduced to help collect machine metadata and assign them slot numbers. - Finally, if there is any such machine metadata, 'MIRPrinter' outputs an additional 'machineMetadataNodes' field containing all the definition of those nodes. Reviewed By: arsenm Differential Revision: https://reviews.llvm.org/D103205
82 lines
3.0 KiB
C++
82 lines
3.0 KiB
C++
//===-- llvm/CodeGen/MachineModuleInfo.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 "llvm/CodeGen/MachineModuleSlotTracker.h"
|
|
#include "llvm/CodeGen/MachineFunction.h"
|
|
#include "llvm/CodeGen/MachineModuleInfo.h"
|
|
|
|
using namespace llvm;
|
|
|
|
void MachineModuleSlotTracker::processMachineFunctionMetadata(
|
|
AbstractSlotTrackerStorage *AST, const MachineFunction &MF) {
|
|
// Create metadata created within the backend.
|
|
for (const MachineBasicBlock &MBB : MF)
|
|
for (const MachineInstr &MI : MBB.instrs())
|
|
for (const MachineMemOperand *MMO : MI.memoperands()) {
|
|
AAMDNodes AAInfo = MMO->getAAInfo();
|
|
if (AAInfo.TBAA)
|
|
AST->createMetadataSlot(AAInfo.TBAA);
|
|
if (AAInfo.TBAAStruct)
|
|
AST->createMetadataSlot(AAInfo.TBAAStruct);
|
|
if (AAInfo.Scope)
|
|
AST->createMetadataSlot(AAInfo.Scope);
|
|
if (AAInfo.NoAlias)
|
|
AST->createMetadataSlot(AAInfo.NoAlias);
|
|
}
|
|
}
|
|
|
|
void MachineModuleSlotTracker::processMachineModule(
|
|
AbstractSlotTrackerStorage *AST, const Module *M,
|
|
bool ShouldInitializeAllMetadata) {
|
|
if (ShouldInitializeAllMetadata) {
|
|
for (const Function &F : *M) {
|
|
if (&F != &TheFunction)
|
|
continue;
|
|
MDNStartSlot = AST->getNextMetadataSlot();
|
|
if (auto *MF = TheMMI.getMachineFunction(F))
|
|
processMachineFunctionMetadata(AST, *MF);
|
|
MDNEndSlot = AST->getNextMetadataSlot();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void MachineModuleSlotTracker::processMachineFunction(
|
|
AbstractSlotTrackerStorage *AST, const Function *F,
|
|
bool ShouldInitializeAllMetadata) {
|
|
if (!ShouldInitializeAllMetadata && F == &TheFunction) {
|
|
MDNStartSlot = AST->getNextMetadataSlot();
|
|
if (auto *MF = TheMMI.getMachineFunction(*F))
|
|
processMachineFunctionMetadata(AST, *MF);
|
|
MDNEndSlot = AST->getNextMetadataSlot();
|
|
}
|
|
}
|
|
|
|
void MachineModuleSlotTracker::collectMachineMDNodes(
|
|
MachineMDNodeListType &L) const {
|
|
collectMDNodes(L, MDNStartSlot, MDNEndSlot);
|
|
}
|
|
|
|
MachineModuleSlotTracker::MachineModuleSlotTracker(
|
|
const MachineFunction *MF, bool ShouldInitializeAllMetadata)
|
|
: ModuleSlotTracker(MF->getFunction().getParent(),
|
|
ShouldInitializeAllMetadata),
|
|
TheFunction(MF->getFunction()), TheMMI(MF->getMMI()), MDNStartSlot(0),
|
|
MDNEndSlot(0) {
|
|
setProcessHook([this](AbstractSlotTrackerStorage *AST, const Module *M,
|
|
bool ShouldInitializeAllMetadata) {
|
|
this->processMachineModule(AST, M, ShouldInitializeAllMetadata);
|
|
});
|
|
setProcessHook([this](AbstractSlotTrackerStorage *AST, const Function *F,
|
|
bool ShouldInitializeAllMetadata) {
|
|
this->processMachineFunction(AST, F, ShouldInitializeAllMetadata);
|
|
});
|
|
}
|
|
|
|
MachineModuleSlotTracker::~MachineModuleSlotTracker() = default;
|