aboutsummaryrefslogtreecommitdiff
path: root/llvm/lib/CodeGen/MachineModuleSlotTracker.cpp
diff options
context:
space:
mode:
authorMichael Liao <michael.hliao@gmail.com>2021-05-25 20:20:52 -0400
committerMichael Liao <michael.hliao@gmail.com>2021-06-19 12:48:08 -0400
commitb9c05aff205bab3f8ca639e44b825277d6cf48a9 (patch)
treeb708bae3e3d795b8ab453ff83e6b535f0f95acd3 /llvm/lib/CodeGen/MachineModuleSlotTracker.cpp
parent940efa4f6981964c95a69ff412378722438607bf (diff)
downloadllvm-b9c05aff205bab3f8ca639e44b825277d6cf48a9.zip
llvm-b9c05aff205bab3f8ca639e44b825277d6cf48a9.tar.gz
llvm-b9c05aff205bab3f8ca639e44b825277d6cf48a9.tar.bz2
[MIRPrinter] Add machine metadata support.
- 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
Diffstat (limited to 'llvm/lib/CodeGen/MachineModuleSlotTracker.cpp')
-rw-r--r--llvm/lib/CodeGen/MachineModuleSlotTracker.cpp81
1 files changed, 81 insertions, 0 deletions
diff --git a/llvm/lib/CodeGen/MachineModuleSlotTracker.cpp b/llvm/lib/CodeGen/MachineModuleSlotTracker.cpp
new file mode 100644
index 0000000..e4da179
--- /dev/null
+++ b/llvm/lib/CodeGen/MachineModuleSlotTracker.cpp
@@ -0,0 +1,81 @@
+//===-- 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;