[llvm] a9adb62 - [AsmPrinter] Use getMnemonic for instruction-mix remark.
Florian Hahn via llvm-commits
llvm-commits at lists.llvm.org
Tue Nov 17 04:13:14 PST 2020
Author: Florian Hahn
Date: 2020-11-17T12:12:47Z
New Revision: a9adb62a640621814d528691b76964209505ebb8
URL: https://github.com/llvm/llvm-project/commit/a9adb62a640621814d528691b76964209505ebb8
DIFF: https://github.com/llvm/llvm-project/commit/a9adb62a640621814d528691b76964209505ebb8.diff
LOG: [AsmPrinter] Use getMnemonic for instruction-mix remark.
This patch uses the new `getMnemonic` helper from D90039
to display mnemonics instead of the internal opcodes.
The main motivation behind using the mnemonics is that they
are more user-friendly and more directly related to the assembly
the users will be presented.
Reviewed By: paquette
Differential Revision: https://reviews.llvm.org/D90040
Added:
Modified:
llvm/include/llvm/MC/MCStreamer.h
llvm/lib/CodeGen/AsmPrinter/AsmPrinter.cpp
llvm/lib/MC/MCAsmStreamer.cpp
llvm/test/CodeGen/AArch64/arm64-instruction-mix-remarks.ll
Removed:
################################################################################
diff --git a/llvm/include/llvm/MC/MCStreamer.h b/llvm/include/llvm/MC/MCStreamer.h
index d90241d89535..6a0ef5360cd4 100644
--- a/llvm/include/llvm/MC/MCStreamer.h
+++ b/llvm/include/llvm/MC/MCStreamer.h
@@ -452,6 +452,10 @@ class MCStreamer {
/// so we can sort on them later.
void AssignFragment(MCSymbol *Symbol, MCFragment *Fragment);
+ /// Returns the mnemonic for \p MI, if the streamer has access to a
+ /// instruction printer and returns an empty string otherwise.
+ virtual StringRef getMnemonic(MCInst &MI) { return ""; }
+
/// Emit a label for \p Symbol into the current section.
///
/// This corresponds to an assembler statement such as:
diff --git a/llvm/lib/CodeGen/AsmPrinter/AsmPrinter.cpp b/llvm/lib/CodeGen/AsmPrinter/AsmPrinter.cpp
index b9f98911678b..e76d4044b572 100644
--- a/llvm/lib/CodeGen/AsmPrinter/AsmPrinter.cpp
+++ b/llvm/lib/CodeGen/AsmPrinter/AsmPrinter.cpp
@@ -1152,11 +1152,10 @@ void AsmPrinter::emitFunctionBody() {
int NumInstsInFunction = 0;
bool CanDoExtraAnalysis = ORE->allowExtraAnalysis(DEBUG_TYPE);
- const TargetInstrInfo *TII = MF->getSubtarget().getInstrInfo();
for (auto &MBB : *MF) {
// Print a label for the basic block.
emitBasicBlockStart(MBB);
- DenseMap<unsigned, unsigned> OpcodeCounts;
+ DenseMap<StringRef, unsigned> MnemonicCounts;
for (auto &MI : MBB) {
// Print the assembly for the instruction.
if (!MI.isPosition() && !MI.isImplicitDef() && !MI.isKill() &&
@@ -1220,7 +1219,10 @@ void AsmPrinter::emitFunctionBody() {
default:
emitInstruction(&MI);
if (CanDoExtraAnalysis) {
- auto I = OpcodeCounts.insert({MI.getOpcode(), 0u});
+ MCInst MCI;
+ MCI.setOpcode(MI.getOpcode());
+ auto Name = OutStreamer->getMnemonic(MCI);
+ auto I = MnemonicCounts.insert({Name, 0u});
I.first->second++;
}
break;
@@ -1272,24 +1274,25 @@ void AsmPrinter::emitFunctionBody() {
MachineOptimizationRemarkAnalysis R(DEBUG_TYPE, "InstructionMix",
MBB.begin()->getDebugLoc(), &MBB);
- // Generate instruction mix remark. First, convert opcodes to string
- // names, then sort them in descending order by count and name.
- SmallVector<std::pair<std::string, unsigned>, 128> OpcodeCountsVec;
- for (auto &KV : OpcodeCounts) {
- auto Name = (Twine("INST_") + TII->getName(KV.first)).str();
- OpcodeCountsVec.emplace_back(Name, KV.second);
- }
- sort(OpcodeCountsVec, [](const std::pair<std::string, unsigned> &A,
- const std::pair<std::string, unsigned> &B) {
+ // Generate instruction mix remark. First, sort counts in descending order
+ // by count and name.
+ SmallVector<std::pair<StringRef, unsigned>, 128> MnemonicVec;
+ for (auto &KV : MnemonicCounts)
+ MnemonicVec.emplace_back(KV.first, KV.second);
+
+ sort(MnemonicVec, [](const std::pair<StringRef, unsigned> &A,
+ const std::pair<StringRef, unsigned> &B) {
if (A.second > B.second)
return true;
if (A.second == B.second)
- return A.first < B.first;
+ return StringRef(A.first) < StringRef(B.first);
return false;
});
R << "BasicBlock: " << ore::NV("BasicBlock", MBB.getName()) << "\n";
- for (auto &KV : OpcodeCountsVec)
- R << KV.first << ": " << ore::NV(KV.first, KV.second) << "\n";
+ for (auto &KV : MnemonicVec) {
+ auto Name = (Twine("INST_") + KV.first.trim()).str();
+ R << KV.first << ": " << ore::NV(Name, KV.second) << "\n";
+ }
ORE->emit(R);
}
}
diff --git a/llvm/lib/MC/MCAsmStreamer.cpp b/llvm/lib/MC/MCAsmStreamer.cpp
index 8d96935b2205..32340b577411 100644
--- a/llvm/lib/MC/MCAsmStreamer.cpp
+++ b/llvm/lib/MC/MCAsmStreamer.cpp
@@ -144,6 +144,11 @@ class MCAsmStreamer final : public MCStreamer {
const MCSymbol *Aliasee) override;
void emitLOHDirective(MCLOHType Kind, const MCLOHArgs &Args) override;
+
+ StringRef getMnemonic(MCInst &MI) override {
+ return InstPrinter->getMnemonic(&MI).first;
+ }
+
void emitLabel(MCSymbol *Symbol, SMLoc Loc = SMLoc()) override;
void emitAssemblerFlag(MCAssemblerFlag Flag) override;
diff --git a/llvm/test/CodeGen/AArch64/arm64-instruction-mix-remarks.ll b/llvm/test/CodeGen/AArch64/arm64-instruction-mix-remarks.ll
index f56d0d2d0372..b722c3006588 100644
--- a/llvm/test/CodeGen/AArch64/arm64-instruction-mix-remarks.ll
+++ b/llvm/test/CodeGen/AArch64/arm64-instruction-mix-remarks.ll
@@ -3,19 +3,24 @@
; CHECK-LABEL: %entry
; CHECK-NEXT: ldr w8, [x0]
+; CHECK-NEXT: mov w10, #16959
+; CHECK-NEXT: movk w10, #15, lsl #16
; CHECK-NEXT: add w8, w8, w1
-; CHECK-NEXT: cmp w8, #100
+; CHECK-NEXT: add x9, x8, x2
+; CHECK-NEXT: cmp x9, x10
; CHECK-NEXT: b.ne LBB0_2
; YAML: Name: InstructionMix
; YAML-NEXT: DebugLoc: { File: arm64-instruction-mix-remarks.ll, Line: 10, Column: 10 }
; YAML-NEXT: Function: foo
; YAML-NEXT: Args:
-; YAML: - BasicBlock: entry
-; YAML: - INST_ADDWrs: '1'
-; YAML: - INST_Bcc: '1'
-; YAML: - INST_LDRWui: '1'
-; YAML: - INST_SUBSWri: '1'
+; YAML: - BasicBlock: entry
+; YAML: - INST_add: '2'
+; YAML: - INST_b.: '1'
+; YAML: - INST_ldr: '1'
+; YAML: - INST_movk: '1'
+; YAML: - INST_movz: '1'
+; YAML: - INST_subs: '1'
; CHECK-LABEL: %then
@@ -26,9 +31,9 @@
; YAML-NEXT: DebugLoc: { File: arm64-instruction-mix-remarks.ll, Line: 20, Column: 20 }
; YAML-NEXT: Function: foo
; YAML-NEXT: Args:
-; YAML: - BasicBlock: then
-; YAML: - INST_ORRWrs: '1'
-; YAML: - INST_RET: '1'
+; YAML: - BasicBlock: then
+; YAML: - INST_orr: '1'
+; YAML: - INST_ret: '1'
; CHECK-LABEL: %else
; CHECK-NEXT: mul w8, w8, w1
@@ -42,16 +47,18 @@
; YAML-NEXT: DebugLoc: { File: arm64-instruction-mix-remarks.ll, Line: 30, Column: 30 }
; YAML-NEXT: Function: foo
; YAML-NEXT: Args:
-; YAML: - BasicBlock: else
-; YAML: - INST_MADDWrrr: '2'
-; YAML: - INST_ORRWrs: '1'
-; YAML: - INST_RET: '1'
-; YAML: - INST_STRWui: '1'
-define i32 @foo(i32* %ptr, i32 %x) !dbg !3 {
+; YAML: - BasicBlock: else
+; YAML: - INST_madd: '2'
+; YAML: - INST_orr: '1'
+; YAML: - INST_ret: '1'
+; YAML: - INST_str: '1'
+define i32 @foo(i32* %ptr, i32 %x, i64 %y) !dbg !3 {
entry:
%l = load i32, i32* %ptr, !dbg !4
%add = add i32 %l, %x, !dbg !4
- %c = icmp eq i32 %add, 100, !dbg !4
+ %add.ext = zext i32 %add to i64, !dbg !4
+ %add.64 = add i64 %add.ext, %y, !dbg !4
+ %c = icmp eq i64 %add.64, 999999, !dbg !4
br i1 %c, label %then, label %else, !dbg !4
then:
More information about the llvm-commits
mailing list