LLVM 19.0.0git
Macros | Functions
WebAssemblyRegStackify.cpp File Reference

This file implements a register stacking pass. More...

#include "MCTargetDesc/WebAssemblyMCTargetDesc.h"
#include "WebAssembly.h"
#include "WebAssemblyDebugValueManager.h"
#include "WebAssemblyMachineFunctionInfo.h"
#include "WebAssemblySubtarget.h"
#include "WebAssemblyUtilities.h"
#include "llvm/Analysis/AliasAnalysis.h"
#include "llvm/CodeGen/LiveIntervals.h"
#include "llvm/CodeGen/MachineBlockFrequencyInfo.h"
#include "llvm/CodeGen/MachineDominators.h"
#include "llvm/CodeGen/MachineInstrBuilder.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"
#include <iterator>

Go to the source code of this file.

Macros

#define DEBUG_TYPE   "wasm-reg-stackify"
 

Functions

 INITIALIZE_PASS (WebAssemblyRegStackify, DEBUG_TYPE, "Reorder instructions to use the WebAssembly value stack", false, false) FunctionPass *llvm
 
static void imposeStackOrdering (MachineInstr *MI)
 
static void convertImplicitDefToConstZero (MachineInstr *MI, MachineRegisterInfo &MRI, const TargetInstrInfo *TII, MachineFunction &MF, LiveIntervals &LIS)
 
static void queryCallee (const MachineInstr &MI, bool &Read, bool &Write, bool &Effects, bool &StackPointer)
 
static void query (const MachineInstr &MI, bool &Read, bool &Write, bool &Effects, bool &StackPointer)
 
static bool shouldRematerialize (const MachineInstr &Def, const WebAssemblyInstrInfo *TII)
 
static MachineInstrgetVRegDef (unsigned Reg, const MachineInstr *Insert, const MachineRegisterInfo &MRI, const LiveIntervals &LIS)
 
static bool hasOneNonDBGUse (unsigned Reg, MachineInstr *Def, MachineRegisterInfo &MRI, MachineDominatorTree &MDT, LiveIntervals &LIS)
 
static bool isSafeToMove (const MachineOperand *Def, const MachineOperand *Use, const MachineInstr *Insert, const WebAssemblyFunctionInfo &MFI, const MachineRegisterInfo &MRI)
 
static bool oneUseDominatesOtherUses (unsigned Reg, const MachineOperand &OneUse, const MachineBasicBlock &MBB, const MachineRegisterInfo &MRI, const MachineDominatorTree &MDT, LiveIntervals &LIS, WebAssemblyFunctionInfo &MFI)
 Test whether OneUse, a use of Reg, dominates all of Reg's other uses.
 
static unsigned getTeeOpcode (const TargetRegisterClass *RC)
 Get the appropriate tee opcode for the given register class.
 
static void shrinkToUses (LiveInterval &LI, LiveIntervals &LIS)
 
static MachineInstrmoveForSingleUse (unsigned Reg, MachineOperand &Op, MachineInstr *Def, MachineBasicBlock &MBB, MachineInstr *Insert, LiveIntervals &LIS, WebAssemblyFunctionInfo &MFI, MachineRegisterInfo &MRI)
 A single-use def in the same block with no intervening memory or register dependencies; move the def down and nest it with the current instruction.
 
static MachineInstrgetPrevNonDebugInst (MachineInstr *MI)
 
static MachineInstrrematerializeCheapDef (unsigned Reg, MachineOperand &Op, MachineInstr &Def, MachineBasicBlock &MBB, MachineBasicBlock::instr_iterator Insert, LiveIntervals &LIS, WebAssemblyFunctionInfo &MFI, MachineRegisterInfo &MRI, const WebAssemblyInstrInfo *TII, const WebAssemblyRegisterInfo *TRI)
 A trivially cloneable instruction; clone it and nest the new copy with the current instruction.
 
static MachineInstrmoveAndTeeForMultiUse (unsigned Reg, MachineOperand &Op, MachineInstr *Def, MachineBasicBlock &MBB, MachineInstr *Insert, LiveIntervals &LIS, WebAssemblyFunctionInfo &MFI, MachineRegisterInfo &MRI, const WebAssemblyInstrInfo *TII)
 A multiple-use def in the same block with no intervening memory or register dependencies; move the def down, nest it with the current instruction, and insert a tee to satisfy the rest of the uses.
 

Detailed Description

This file implements a register stacking pass.

This pass reorders instructions to put register uses and defs in an order such that they form single-use expression trees. Registers fitting this form are then marked as "stackified", meaning references to them are replaced by "push" and "pop" from the value stack.

This is primarily a code size optimization, since temporary values on the value stack don't need to be named.

Definition in file WebAssemblyRegStackify.cpp.

Macro Definition Documentation

◆ DEBUG_TYPE

#define DEBUG_TYPE   "wasm-reg-stackify"

Definition at line 41 of file WebAssemblyRegStackify.cpp.

Function Documentation

◆ convertImplicitDefToConstZero()

static void convertImplicitDefToConstZero ( MachineInstr MI,
MachineRegisterInfo MRI,
const TargetInstrInfo TII,
MachineFunction MF,
LiveIntervals LIS 
)
static

◆ getPrevNonDebugInst()

static MachineInstr * getPrevNonDebugInst ( MachineInstr MI)
static

Definition at line 560 of file WebAssemblyRegStackify.cpp.

References I, and MI.

Referenced by rematerializeCheapDef().

◆ getTeeOpcode()

static unsigned getTeeOpcode ( const TargetRegisterClass RC)
static

Get the appropriate tee opcode for the given register class.

Definition at line 492 of file WebAssemblyRegStackify.cpp.

References llvm_unreachable.

Referenced by moveAndTeeForMultiUse().

◆ getVRegDef()

static MachineInstr * getVRegDef ( unsigned  Reg,
const MachineInstr Insert,
const MachineRegisterInfo MRI,
const LiveIntervals LIS 
)
static

◆ hasOneNonDBGUse()

static bool hasOneNonDBGUse ( unsigned  Reg,
MachineInstr Def,
MachineRegisterInfo MRI,
MachineDominatorTree MDT,
LiveIntervals LIS 
)
static

◆ imposeStackOrdering()

static void imposeStackOrdering ( MachineInstr MI)
static

◆ INITIALIZE_PASS()

INITIALIZE_PASS ( WebAssemblyRegStackify  ,
DEBUG_TYPE  ,
"Reorder instructions to use the WebAssembly value stack"  ,
false  ,
false   
)

Definition at line 70 of file WebAssemblyRegStackify.cpp.

◆ isSafeToMove()

static bool isSafeToMove ( const MachineOperand Def,
const MachineOperand Use,
const MachineInstr Insert,
const WebAssemblyFunctionInfo MFI,
const MachineRegisterInfo MRI 
)
static

◆ moveAndTeeForMultiUse()

static MachineInstr * moveAndTeeForMultiUse ( unsigned  Reg,
MachineOperand Op,
MachineInstr Def,
MachineBasicBlock MBB,
MachineInstr Insert,
LiveIntervals LIS,
WebAssemblyFunctionInfo MFI,
MachineRegisterInfo MRI,
const WebAssemblyInstrInfo TII 
)
static

A multiple-use def in the same block with no intervening memory or register dependencies; move the def down, nest it with the current instruction, and insert a tee to satisfy the rest of the uses.

As an illustration, rewrite this:

Reg = INST ... // Def INST ..., Reg, ... // Insert INST ..., Reg, ... INST ..., Reg, ...

to this:

DefReg = INST ... // Def (to become the new Insert) TeeReg, Reg = TEE_... DefReg INST ..., TeeReg, ... // Insert INST ..., Reg, ... INST ..., Reg, ...

with DefReg and TeeReg stackified. This eliminates a local.get from the resulting code.

Definition at line 632 of file WebAssemblyRegStackify.cpp.

References llvm::MachineInstrBuilder::addReg(), llvm::BuildMI(), llvm::WebAssemblyDebugValueManager::cloneSink(), llvm::LiveIntervals::createAndComputeVirtRegInterval(), llvm::dbgs(), llvm::VNInfo::def, llvm::RegState::Define, llvm::MachineInstr::dump(), llvm::LiveRange::FindSegmentContaining(), llvm::LiveIntervals::getInstructionIndex(), llvm::LiveIntervals::getInterval(), llvm::SlotIndex::getRegSlot(), getTeeOpcode(), llvm::getUndefRegState(), llvm::LiveRange::getVNInfoAt(), llvm::LiveIntervals::handleMove(), I, imposeStackOrdering(), llvm::LiveIntervals::InsertMachineInstrInMaps(), llvm::MachineOperand::isDead(), LLVM_DEBUG, MBB, MRI, shrinkToUses(), llvm::WebAssemblyDebugValueManager::sink(), llvm::WebAssemblyFunctionInfo::stackifyVReg(), TII, and llvm::WebAssemblyDebugValueManager::updateReg().

◆ moveForSingleUse()

static MachineInstr * moveForSingleUse ( unsigned  Reg,
MachineOperand Op,
MachineInstr Def,
MachineBasicBlock MBB,
MachineInstr Insert,
LiveIntervals LIS,
WebAssemblyFunctionInfo MFI,
MachineRegisterInfo MRI 
)
static

◆ oneUseDominatesOtherUses()

static bool oneUseDominatesOtherUses ( unsigned  Reg,
const MachineOperand OneUse,
const MachineBasicBlock MBB,
const MachineRegisterInfo MRI,
const MachineDominatorTree MDT,
LiveIntervals LIS,
WebAssemblyFunctionInfo MFI 
)
static

◆ query()

static void query ( const MachineInstr MI,
bool Read,
bool Write,
bool Effects,
bool StackPointer 
)
static

Definition at line 165 of file WebAssemblyRegStackify.cpp.

References assert(), MI, and queryCallee().

Referenced by isSafeToMove().

◆ queryCallee()

static void queryCallee ( const MachineInstr MI,
bool Read,
bool Write,
bool Effects,
bool StackPointer 
)
static

◆ rematerializeCheapDef()

static MachineInstr * rematerializeCheapDef ( unsigned  Reg,
MachineOperand Op,
MachineInstr Def,
MachineBasicBlock MBB,
MachineBasicBlock::instr_iterator  Insert,
LiveIntervals LIS,
WebAssemblyFunctionInfo MFI,
MachineRegisterInfo MRI,
const WebAssemblyInstrInfo TII,
const WebAssemblyRegisterInfo TRI 
)
static

◆ shouldRematerialize()

static bool shouldRematerialize ( const MachineInstr Def,
const WebAssemblyInstrInfo TII 
)
static

Definition at line 256 of file WebAssemblyRegStackify.cpp.

References TII.

◆ shrinkToUses()

static void shrinkToUses ( LiveInterval LI,
LiveIntervals LIS 
)
static