Skip to content

Commit 35cffa7

Browse files
toppercAmanieu
authored andcommitted
[X86] Add x, t and g modifiers for inline asm
This patch adds the x, t and g modifiers for inline asm from GCC. These will print a vector register as xmm*, ymm* or zmm* respectively. I also fixed register names with modifiers with inteldialect so they are no longer printed with a leading %. Patch by Amanieu d'Antras Differential Revision: https://reviews.llvm.org/D78977
1 parent 6a831f0 commit 35cffa7

File tree

3 files changed

+100
-3
lines changed

3 files changed

+100
-3
lines changed

llvm/lib/Target/X86/X86AsmPrinter.cpp

+45-1
Original file line numberDiff line numberDiff line change
@@ -404,7 +404,7 @@ void X86AsmPrinter::PrintIntelMemReference(const MachineInstr *MI,
404404
static bool printAsmMRegister(X86AsmPrinter &P, const MachineOperand &MO,
405405
char Mode, raw_ostream &O) {
406406
Register Reg = MO.getReg();
407-
bool EmitPercent = true;
407+
bool EmitPercent = MO.getParent()->getInlineAsmDialect() == InlineAsm::AD_ATT;
408408

409409
if (!X86::GR8RegClass.contains(Reg) &&
410410
!X86::GR16RegClass.contains(Reg) &&
@@ -443,6 +443,42 @@ static bool printAsmMRegister(X86AsmPrinter &P, const MachineOperand &MO,
443443
return false;
444444
}
445445

446+
static bool printAsmVRegister(X86AsmPrinter &P, const MachineOperand &MO,
447+
char Mode, raw_ostream &O) {
448+
unsigned Reg = MO.getReg();
449+
bool EmitPercent = MO.getParent()->getInlineAsmDialect() == InlineAsm::AD_ATT;
450+
451+
unsigned Index;
452+
if (X86::VR128XRegClass.contains(Reg))
453+
Index = Reg - X86::XMM0;
454+
else if (X86::VR256XRegClass.contains(Reg))
455+
Index = Reg - X86::YMM0;
456+
else if (X86::VR512RegClass.contains(Reg))
457+
Index = Reg - X86::ZMM0;
458+
else
459+
return true;
460+
461+
switch (Mode) {
462+
default: // Unknown mode.
463+
return true;
464+
case 'x': // Print V4SFmode register
465+
Reg = X86::XMM0 + Index;
466+
break;
467+
case 't': // Print V8SFmode register
468+
Reg = X86::YMM0 + Index;
469+
break;
470+
case 'g': // Print V16SFmode register
471+
Reg = X86::ZMM0 + Index;
472+
break;
473+
}
474+
475+
if (EmitPercent)
476+
O << '%';
477+
478+
O << X86ATTInstPrinter::getRegisterName(Reg);
479+
return false;
480+
}
481+
446482
/// PrintAsmOperand - Print out an operand for an inline asm expression.
447483
///
448484
bool X86AsmPrinter::PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
@@ -517,6 +553,14 @@ bool X86AsmPrinter::PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
517553
PrintOperand(MI, OpNo, O);
518554
return false;
519555

556+
case 'x': // Print V4SFmode register
557+
case 't': // Print V8SFmode register
558+
case 'g': // Print V16SFmode register
559+
if (MO.isReg())
560+
return printAsmVRegister(*this, MO, ExtraCode[0], O);
561+
PrintOperand(MI, OpNo, O);
562+
return false;
563+
520564
case 'P': // This is the operand of a call, treat specially.
521565
PrintPCRelImm(MI, OpNo, O);
522566
return false;
+53
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,53 @@
1+
; RUN: llc < %s | FileCheck %s
2+
3+
define void @test1() {
4+
; CHECK-LABEL: test1:
5+
; CHECK: vmovaps %xmm0, %xmm0
6+
; CHECK: vmovaps %ymm0, %ymm0
7+
; CHECK: vmovaps %zmm0, %zmm0
8+
tail call void asm sideeffect "vmovaps ${0:x}, ${0:x}", "{xmm0},~{dirflag},~{fpsr},~{flags}"(i32 0)
9+
tail call void asm sideeffect "vmovaps ${0:t}, ${0:t}", "{xmm0},~{dirflag},~{fpsr},~{flags}"(i32 0)
10+
tail call void asm sideeffect "vmovaps ${0:g}, ${0:g}", "{xmm0},~{dirflag},~{fpsr},~{flags}"(i32 0)
11+
ret void
12+
}
13+
14+
define void @test2() {
15+
; CHECK-LABEL: test2:
16+
; CHECK: vmovaps %xmm0, %xmm0
17+
; CHECK: vmovaps %ymm0, %ymm0
18+
; CHECK: vmovaps %zmm0, %zmm0
19+
tail call void asm sideeffect inteldialect "vmovaps ${0:x}, ${0:x}", "{xmm0},~{dirflag},~{fpsr},~{flags}"(i32 0)
20+
tail call void asm sideeffect inteldialect "vmovaps ${0:t}, ${0:t}", "{xmm0},~{dirflag},~{fpsr},~{flags}"(i32 0)
21+
tail call void asm sideeffect inteldialect "vmovaps ${0:g}, ${0:g}", "{xmm0},~{dirflag},~{fpsr},~{flags}"(i32 0)
22+
ret void
23+
}
24+
25+
define void @test3() {
26+
; CHECK-LABEL: test3:
27+
; CHECK: movb %al, %al
28+
; CHECK: movb %ah, %ah
29+
; CHECK: movw %ax, %ax
30+
; CHECK: movl %eax, %eax
31+
; CHECK: movq %rax, %rax
32+
tail call void asm sideeffect "mov ${0:b}, ${0:b}", "{eax},~{dirflag},~{fpsr},~{flags}"(i32 0)
33+
tail call void asm sideeffect "mov ${0:h}, ${0:h}", "{eax},~{dirflag},~{fpsr},~{flags}"(i32 0)
34+
tail call void asm sideeffect "mov ${0:w}, ${0:w}", "{eax},~{dirflag},~{fpsr},~{flags}"(i32 0)
35+
tail call void asm sideeffect "mov ${0:k}, ${0:k}", "{eax},~{dirflag},~{fpsr},~{flags}"(i32 0)
36+
tail call void asm sideeffect "mov ${0:q}, ${0:q}", "{eax},~{dirflag},~{fpsr},~{flags}"(i32 0)
37+
ret void
38+
}
39+
40+
define void @test4() {
41+
; CHECK-LABEL: test4:
42+
; CHECK: movb %al, %al
43+
; CHECK: movb %ah, %ah
44+
; CHECK: movw %ax, %ax
45+
; CHECK: movl %eax, %eax
46+
; CHECK: movq %rax, %rax
47+
tail call void asm sideeffect inteldialect "mov ${0:b}, ${0:b}", "{eax},~{dirflag},~{fpsr},~{flags}"(i32 0)
48+
tail call void asm sideeffect inteldialect "mov ${0:h}, ${0:h}", "{eax},~{dirflag},~{fpsr},~{flags}"(i32 0)
49+
tail call void asm sideeffect inteldialect "mov ${0:w}, ${0:w}", "{eax},~{dirflag},~{fpsr},~{flags}"(i32 0)
50+
tail call void asm sideeffect inteldialect "mov ${0:k}, ${0:k}", "{eax},~{dirflag},~{fpsr},~{flags}"(i32 0)
51+
tail call void asm sideeffect inteldialect "mov ${0:q}, ${0:q}", "{eax},~{dirflag},~{fpsr},~{flags}"(i32 0)
52+
ret void
53+
}
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
; RUN: not llc -mtriple=x86_64-- < %s 2>&1 | FileCheck %s
22

3-
;CHECK: error: invalid operand in inline asm: 'vmovd ${1:x}, $0'
3+
;CHECK: error: invalid operand in inline asm: 'vmovd ${1:k}, $0'
44
define i32 @foo() {
55
entry:
6-
%0 = tail call i32 asm sideeffect "vmovd ${1:x}, $0", "=r,x,~{dirflag},~{fpsr},~{flags}"(<2 x i64> <i64 240518168632, i64 240518168632>)
6+
%0 = tail call i32 asm sideeffect "vmovd ${1:k}, $0", "=r,x,~{dirflag},~{fpsr},~{flags}"(<2 x i64> <i64 240518168632, i64 240518168632>)
77
ret i32 %0
88
}

0 commit comments

Comments
 (0)