xref: /llvm-project/llvm/test/CodeGen/X86/stack-folding-bmi2.mir (revision 60442f0d442723a487528bdd8b48b24657a025e8) 
1# NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
2# RUN: llc -o - -mtriple=x86_64-- -run-pass=greedy %s | FileCheck %s
3# Tests for stack folding MULX. Doing this as a MIR test to ensure MULX is used and not legacy MUL.
4--- |
5  ; Function Attrs: nounwind
6  define i32 @stack_fold_mulx_u32(i32 %a0, i32 %a1) #0 {
7    tail call void asm sideeffect "nop", "~{rax},~{rbx},~{rcx},~{rdx},~{rsi},~{rdi},~{rbp},~{r8},~{r9},~{r10},~{r11},~{r12},~{r13},~{r14},~{r15}"()
8    %1 = zext i32 %a0 to i128
9    %2 = zext i32 %a1 to i128
10    %3 = mul i128 %1, %2
11    %4 = lshr i128 %3, 32
12    %5 = trunc i128 %4 to i32
13    ret i32 %5
14  }
15
16  ; Function Attrs: nounwind
17  define i64 @stack_fold_mulx_u64(i64 %a0, i64 %a1) #0 {
18    tail call void asm sideeffect "nop", "~{rax},~{rbx},~{rcx},~{rdx},~{rsi},~{rdi},~{rbp},~{r8},~{r9},~{r10},~{r11},~{r12},~{r13},~{r14},~{r15}"()
19    %1 = zext i64 %a0 to i128
20    %2 = zext i64 %a1 to i128
21    %3 = mul i128 %1, %2
22    %4 = lshr i128 %3, 64
23    %5 = trunc i128 %4 to i64
24    ret i64 %5
25  }
26
27  ; Function Attrs: nounwind
28  declare void @llvm.stackprotector(ptr, ptr) #1
29
30  attributes #0 = { nounwind "target-features"="+bmi2" }
31  attributes #1 = { nounwind }
32
33...
34---
35name:            stack_fold_mulx_u32
36alignment:       16
37tracksRegLiveness: true
38registers:
39  - { id: 0, class: gr32 }
40  - { id: 1, class: gr32 }
41  - { id: 2, class: fr64 }
42  - { id: 3, class: gr32 }
43  - { id: 4, class: gr32 }
44liveins:
45  - { reg: '$edi', virtual-reg: '%0' }
46  - { reg: '$esi', virtual-reg: '%1' }
47body:             |
48  bb.0 (%ir-block.0):
49    liveins: $edi, $esi
50
51    ; CHECK-LABEL: name: stack_fold_mulx_u32
52    ; CHECK: liveins: $edi, $esi
53    ; CHECK: MOV32mr %stack.0, 1, $noreg, 0, $noreg, $esi :: (store (s32) into %stack.0)
54    ; CHECK: MOV32mr %stack.1, 1, $noreg, 0, $noreg, $edi :: (store (s32) into %stack.1)
55    ; CHECK: INLINEASM &nop, 1 /* sideeffect attdialect */, 12 /* clobber */, implicit-def dead early-clobber $rax, 12 /* clobber */, implicit-def dead early-clobber $rbx, 12 /* clobber */, implicit-def dead early-clobber $rcx, 12 /* clobber */, implicit-def dead early-clobber $rdx, 12 /* clobber */, implicit-def dead early-clobber $rsi, 12 /* clobber */, implicit-def dead early-clobber $rdi, 12 /* clobber */, implicit-def dead early-clobber $rbp, 12 /* clobber */, implicit-def dead early-clobber $r8, 12 /* clobber */, implicit-def dead early-clobber $r9, 12 /* clobber */, implicit-def dead early-clobber $r10, 12 /* clobber */, implicit-def dead early-clobber $r11, 12 /* clobber */, implicit-def dead early-clobber $r12, 12 /* clobber */, implicit-def dead early-clobber $r13, 12 /* clobber */, implicit-def dead early-clobber $r14, 12 /* clobber */, implicit-def dead early-clobber $r15
56    ; CHECK: $edx = MOV32rm %stack.1, 1, $noreg, 0, $noreg :: (load (s32) from %stack.1)
57    ; CHECK: %3:gr32, dead %4:gr32 = MULX32rm %stack.0, 1, $noreg, 0, $noreg, implicit $edx :: (load (s32) from %stack.0)
58    ; CHECK: $eax = COPY %3
59    ; CHECK: RET 0, $eax
60    %1:gr32 = COPY $esi
61    %0:gr32 = COPY $edi
62    INLINEASM &nop, 1 /* sideeffect attdialect */, 12 /* clobber */, implicit-def dead early-clobber $rax, 12 /* clobber */, implicit-def dead early-clobber $rbx, 12 /* clobber */, implicit-def dead early-clobber $rcx, 12 /* clobber */, implicit-def dead early-clobber $rdx, 12 /* clobber */, implicit-def dead early-clobber $rsi, 12 /* clobber */, implicit-def dead early-clobber $rdi, 12 /* clobber */, implicit-def dead early-clobber $rbp, 12 /* clobber */, implicit-def dead early-clobber $r8, 12 /* clobber */, implicit-def dead early-clobber $r9, 12 /* clobber */, implicit-def dead early-clobber $r10, 12 /* clobber */, implicit-def dead early-clobber $r11, 12 /* clobber */, implicit-def dead early-clobber $r12, 12 /* clobber */, implicit-def dead early-clobber $r13, 12 /* clobber */, implicit-def dead early-clobber $r14, 12 /* clobber */, implicit-def dead early-clobber $r15
63    $edx = COPY %0
64    %3:gr32, dead %4:gr32 = MULX32rr %1, implicit killed $edx
65    $eax = COPY %3
66    RET 0, killed $eax
67
68...
69---
70name:            stack_fold_mulx_u64
71alignment:       16
72tracksRegLiveness: true
73registers:
74  - { id: 0, class: gr64 }
75  - { id: 1, class: gr64 }
76  - { id: 2, class: fr64 }
77  - { id: 3, class: gr64 }
78  - { id: 4, class: gr64 }
79liveins:
80  - { reg: '$rdi', virtual-reg: '%0' }
81  - { reg: '$rsi', virtual-reg: '%1' }
82body:             |
83  bb.0 (%ir-block.0):
84    liveins: $rdi, $rsi
85
86    ; CHECK-LABEL: name: stack_fold_mulx_u64
87    ; CHECK: liveins: $rdi, $rsi
88    ; CHECK: MOV64mr %stack.0, 1, $noreg, 0, $noreg, $rsi :: (store (s64) into %stack.0)
89    ; CHECK: MOV64mr %stack.1, 1, $noreg, 0, $noreg, $rdi :: (store (s64) into %stack.1)
90    ; CHECK: INLINEASM &nop, 1 /* sideeffect attdialect */, 12 /* clobber */, implicit-def dead early-clobber $rax, 12 /* clobber */, implicit-def dead early-clobber $rbx, 12 /* clobber */, implicit-def dead early-clobber $rcx, 12 /* clobber */, implicit-def dead early-clobber $rdx, 12 /* clobber */, implicit-def dead early-clobber $rsi, 12 /* clobber */, implicit-def dead early-clobber $rdi, 12 /* clobber */, implicit-def dead early-clobber $rbp, 12 /* clobber */, implicit-def dead early-clobber $r8, 12 /* clobber */, implicit-def dead early-clobber $r9, 12 /* clobber */, implicit-def dead early-clobber $r10, 12 /* clobber */, implicit-def dead early-clobber $r11, 12 /* clobber */, implicit-def dead early-clobber $r12, 12 /* clobber */, implicit-def dead early-clobber $r13, 12 /* clobber */, implicit-def dead early-clobber $r14, 12 /* clobber */, implicit-def dead early-clobber $r15
91    ; CHECK: $rdx = MOV64rm %stack.1, 1, $noreg, 0, $noreg :: (load (s64) from %stack.1)
92    ; CHECK: %3:gr64, dead %4:gr64 = MULX64rm %stack.0, 1, $noreg, 0, $noreg, implicit $rdx :: (load (s64) from %stack.0)
93    ; CHECK: $rax = COPY %3
94    ; CHECK: RET 0, $rax
95    %1:gr64 = COPY $rsi
96    %0:gr64 = COPY $rdi
97    INLINEASM &nop, 1 /* sideeffect attdialect */, 12 /* clobber */, implicit-def dead early-clobber $rax, 12 /* clobber */, implicit-def dead early-clobber $rbx, 12 /* clobber */, implicit-def dead early-clobber $rcx, 12 /* clobber */, implicit-def dead early-clobber $rdx, 12 /* clobber */, implicit-def dead early-clobber $rsi, 12 /* clobber */, implicit-def dead early-clobber $rdi, 12 /* clobber */, implicit-def dead early-clobber $rbp, 12 /* clobber */, implicit-def dead early-clobber $r8, 12 /* clobber */, implicit-def dead early-clobber $r9, 12 /* clobber */, implicit-def dead early-clobber $r10, 12 /* clobber */, implicit-def dead early-clobber $r11, 12 /* clobber */, implicit-def dead early-clobber $r12, 12 /* clobber */, implicit-def dead early-clobber $r13, 12 /* clobber */, implicit-def dead early-clobber $r14, 12 /* clobber */, implicit-def dead early-clobber $r15
98    $rdx = COPY %0
99    %3:gr64, dead %4:gr64 = MULX64rr %1, implicit killed $rdx
100    $rax = COPY %3
101    RET 0, killed $rax
102
103...
104