c67c9cfe3f
Reland of 00bf4755. This patches fixes the visibility and linkage information of symbols referring to IR globals. Emission of external declarations is now done in the first execution of emitConstantPool rather than in emitLinkage (and a few other places). This is the point where we have already gathered information about used symbols (by running the MC Lower PrePass) and not yet started emitting any functions so that any declarations that need to be emitted are done so at the top of the file before any functions. This changes the order of a few directives in the final asm file which required an update to a few tests. Reviewed By: sbc100 Differential Revision: https://reviews.llvm.org/D118995
103 lines
3.4 KiB
C++
103 lines
3.4 KiB
C++
//===-- WebAssemblyMCLowerPrePass.cpp - Prepare for MC lower --------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// \file
|
|
/// Some information in MC lowering / asm printing gets generated as
|
|
/// instructions get emitted, but may be necessary at the start, such as for
|
|
/// .globaltype declarations. This pass collects this information.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "MCTargetDesc/WebAssemblyMCTargetDesc.h"
|
|
#include "Utils/WebAssemblyUtilities.h"
|
|
#include "WebAssembly.h"
|
|
#include "WebAssemblyMachineFunctionInfo.h"
|
|
#include "WebAssemblySubtarget.h"
|
|
#include "llvm/ADT/SCCIterator.h"
|
|
#include "llvm/CodeGen/MachineFrameInfo.h"
|
|
#include "llvm/CodeGen/MachineFunction.h"
|
|
#include "llvm/CodeGen/MachineInstrBuilder.h"
|
|
#include "llvm/CodeGen/MachineLoopInfo.h"
|
|
#include "llvm/CodeGen/MachineModuleInfoImpls.h"
|
|
#include "llvm/CodeGen/MachineRegisterInfo.h"
|
|
#include "llvm/CodeGen/Passes.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
using namespace llvm;
|
|
|
|
#define DEBUG_TYPE "wasm-mclower-prepass"
|
|
|
|
namespace {
|
|
class WebAssemblyMCLowerPrePass final : public ModulePass {
|
|
StringRef getPassName() const override {
|
|
return "WebAssembly MC Lower Pre Pass";
|
|
}
|
|
|
|
void getAnalysisUsage(AnalysisUsage &AU) const override {
|
|
AU.setPreservesCFG();
|
|
ModulePass::getAnalysisUsage(AU);
|
|
}
|
|
|
|
bool runOnModule(Module &M) override;
|
|
|
|
public:
|
|
static char ID; // Pass identification, replacement for typeid
|
|
WebAssemblyMCLowerPrePass() : ModulePass(ID) {}
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
char WebAssemblyMCLowerPrePass::ID = 0;
|
|
INITIALIZE_PASS(
|
|
WebAssemblyMCLowerPrePass, DEBUG_TYPE,
|
|
"Collects information ahead of time for MC lowering",
|
|
false, false)
|
|
|
|
ModulePass *llvm::createWebAssemblyMCLowerPrePass() {
|
|
return new WebAssemblyMCLowerPrePass();
|
|
}
|
|
|
|
// NOTE: this is a ModulePass since we need to enforce that this code has run
|
|
// for all functions before AsmPrinter. If this way of doing things is ever
|
|
// suboptimal, we could opt to make it a MachineFunctionPass and instead use
|
|
// something like createBarrierNoopPass() to enforce ordering.
|
|
//
|
|
// The information stored here is essential for emitExternalDecls in the Wasm
|
|
// AsmPrinter
|
|
bool WebAssemblyMCLowerPrePass::runOnModule(Module &M) {
|
|
auto *MMIWP = getAnalysisIfAvailable<MachineModuleInfoWrapperPass>();
|
|
if (!MMIWP)
|
|
return true;
|
|
|
|
MachineModuleInfo &MMI = MMIWP->getMMI();
|
|
MachineModuleInfoWasm &MMIW = MMI.getObjFileInfo<MachineModuleInfoWasm>();
|
|
|
|
for (Function &F : M) {
|
|
MachineFunction *MF = MMI.getMachineFunction(F);
|
|
if (!MF)
|
|
continue;
|
|
|
|
LLVM_DEBUG(dbgs() << "********** MC Lower Pre Pass **********\n"
|
|
"********** Function: "
|
|
<< MF->getName() << '\n');
|
|
|
|
for (MachineBasicBlock &MBB : *MF) {
|
|
for (auto &MI : MBB) {
|
|
// FIXME: what should all be filtered out beyond these?
|
|
if (MI.isDebugInstr() || MI.isInlineAsm())
|
|
continue;
|
|
for (MachineOperand &MO : MI.uses()) {
|
|
if (MO.isSymbol()) {
|
|
MMIW.MachineSymbolsUsed.insert(MO.getSymbolName());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|