diff options
Diffstat (limited to 'llvm/lib')
| -rw-r--r-- | llvm/lib/CodeGen/XRayInstrumentation.cpp | 23 | ||||
| -rw-r--r-- | llvm/lib/Target/PowerPC/PPCAsmPrinter.cpp | 2 |
2 files changed, 9 insertions, 16 deletions
diff --git a/llvm/lib/CodeGen/XRayInstrumentation.cpp b/llvm/lib/CodeGen/XRayInstrumentation.cpp index 0b4c6e55166..e6b65825f5b 100644 --- a/llvm/lib/CodeGen/XRayInstrumentation.cpp +++ b/llvm/lib/CodeGen/XRayInstrumentation.cpp @@ -14,8 +14,8 @@ // //===---------------------------------------------------------------------===// -#include "llvm/ADT/SmallVector.h" #include "llvm/ADT/STLExtras.h" +#include "llvm/ADT/SmallVector.h" #include "llvm/ADT/Triple.h" #include "llvm/CodeGen/MachineBasicBlock.h" #include "llvm/CodeGen/MachineDominators.h" @@ -109,22 +109,13 @@ void XRayInstrumentation::replaceRetWithPatchableRet( void XRayInstrumentation::prependRetWithPatchableExit( MachineFunction &MF, const TargetInstrInfo *TII) { - for (auto &MBB : MF) { - for (auto &T : MBB.terminators()) { - unsigned Opc = 0; + for (auto &MBB : MF) + for (auto &T : MBB.terminators()) if (T.isReturn()) { - Opc = TargetOpcode::PATCHABLE_FUNCTION_EXIT; - } - if (TII->isTailCall(T)) { - Opc = TargetOpcode::PATCHABLE_TAIL_CALL; - } - if (Opc != 0) { - // Prepend the return instruction with PATCHABLE_FUNCTION_EXIT or - // PATCHABLE_TAIL_CALL . - BuildMI(MBB, T, T.getDebugLoc(), TII->get(Opc)); + // Prepend the return instruction with PATCHABLE_FUNCTION_EXIT. + BuildMI(MBB, T, T.getDebugLoc(), + TII->get(TargetOpcode::PATCHABLE_FUNCTION_EXIT)); } - } - } } bool XRayInstrumentation::runOnMachineFunction(MachineFunction &MF) { @@ -143,7 +134,7 @@ bool XRayInstrumentation::runOnMachineFunction(MachineFunction &MF) { // Count the number of MachineInstr`s in MachineFunction int64_t MICount = 0; - for (const auto& MBB : MF) + for (const auto &MBB : MF) MICount += MBB.size(); // Check if we have a loop. diff --git a/llvm/lib/Target/PowerPC/PPCAsmPrinter.cpp b/llvm/lib/Target/PowerPC/PPCAsmPrinter.cpp index 841b8c51446..67c849b37bd 100644 --- a/llvm/lib/Target/PowerPC/PPCAsmPrinter.cpp +++ b/llvm/lib/Target/PowerPC/PPCAsmPrinter.cpp @@ -1131,6 +1131,8 @@ void PPCLinuxAsmPrinter::EmitInstruction(const MachineInstr *MI) { break; } case TargetOpcode::PATCHABLE_TAIL_CALL: + // TODO: Define a trampoline `__xray_FunctionTailExit` and differentiate a + // normal function exit from a tail exit. case TargetOpcode::PATCHABLE_RET: // PPC's tail call instruction, e.g. PPC::TCRETURNdi8, doesn't really // lower to a PPC::B instruction. The PPC::B instruction is generated |
