test/CodeGen/X86/stack-folding-bmi2.mir

   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     %1 = tail call i32 asm sideeffect "nop", "=x,~{rax},~{rbx},~{rcx},~{rdx},~{rsi},~{rdi},~{rbp},~{r8},~{r9},~{r10},~{r11},~{r12},~{r13},~{r14},~{r15}"()
   8     %2 = zext i32 %a0 to i128
   9     %3 = zext i32 %a1 to i128
  10     %4 = mul i128 %2, %3
  11     %5 = lshr i128 %4, 32
  12     %6 = trunc i128 %5 to i32
  13     ret i32 %6
  14   }
  15
  16   ; Function Attrs: nounwind
  17   define i64 @stack_fold_mulx_u64(i64 %a0, i64 %a1) #0 {
  18     %1 = tail call i64 asm sideeffect "nop", "=x,~{rax},~{rbx},~{rcx},~{rdx},~{rsi},~{rdi},~{rbp},~{r8},~{r9},~{r10},~{r11},~{r12},~{r13},~{r14},~{r15}"()
  19     %2 = zext i64 %a0 to i128
  20     %3 = zext i64 %a1 to i128
  21     %4 = mul i128 %2, %3
  22     %5 = lshr i128 %4, 64
  23     %6 = trunc i128 %5 to i64
  24     ret i64 %6
  25   }
  26
  27   ; Function Attrs: nounwind
  28   declare void @llvm.stackprotector(i8*, i8**) #1
  29
  30   attributes #0 = { nounwind "target-features"="+bmi2" }
  31   attributes #1 = { nounwind }
  32
  33 ...
  34 ---
  35 name:            stack_fold_mulx_u32
  36 alignment:       4
  37 tracksRegLiveness: true
  38 registers:
  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 }
  44 liveins:
  45   - { reg: '$edi', virtual-reg: '%0' }
  46   - { reg: '$esi', virtual-reg: '%1' }
  47 body:             |
  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 4 into %stack.0)
  54     ; CHECK: MOV32mr %stack.1, 1, $noreg, 0, $noreg, $edi :: (store 4 into %stack.1)
  55     ; CHECK: INLINEASM &nop, 1, 4063242, def dead %2, 12, implicit-def dead early-clobber $rax, 12, implicit-def dead early-clobber $rbx, 12, implicit-def dead early-clobber $rcx, 12, implicit-def dead early-clobber $rdx, 12, implicit-def dead early-clobber $rsi, 12, implicit-def dead early-clobber $rdi, 12, implicit-def dead early-clobber $rbp, 12, implicit-def dead early-clobber $r8, 12, implicit-def dead early-clobber $r9, 12, implicit-def dead early-clobber $r10, 12, implicit-def dead early-clobber $r11, 12, implicit-def dead early-clobber $r12, 12, implicit-def dead early-clobber $r13, 12, implicit-def dead early-clobber $r14, 12, implicit-def dead early-clobber $r15
  56     ; CHECK: $edx = MOV32rm %stack.1, 1, $noreg, 0, $noreg :: (load 4 from %stack.1)
  57     ; CHECK: %3:gr32, dead %4:gr32 = MULX32rm %stack.0, 1, $noreg, 0, $noreg, implicit $edx :: (load 4 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, 4063242, def dead %2, 12, implicit-def dead early-clobber $rax, 12, implicit-def dead early-clobber $rbx, 12, implicit-def dead early-clobber $rcx, 12, implicit-def dead early-clobber $rdx, 12, implicit-def dead early-clobber $rsi, 12, implicit-def dead early-clobber $rdi, 12, implicit-def dead early-clobber $rbp, 12, implicit-def dead early-clobber $r8, 12, implicit-def dead early-clobber $r9, 12, implicit-def dead early-clobber $r10, 12, implicit-def dead early-clobber $r11, 12, implicit-def dead early-clobber $r12, 12, implicit-def dead early-clobber $r13, 12, implicit-def dead early-clobber $r14, 12, 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 ---
  70 name:            stack_fold_mulx_u64
  71 alignment:       4
  72 tracksRegLiveness: true
  73 registers:
  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 }
  79 liveins:
  80   - { reg: '$rdi', virtual-reg: '%0' }
  81   - { reg: '$rsi', virtual-reg: '%1' }
  82 body:             |
  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 8 into %stack.0)
  89     ; CHECK: MOV64mr %stack.1, 1, $noreg, 0, $noreg, $rdi :: (store 8 into %stack.1)
  90     ; CHECK: INLINEASM &nop, 1, 4063242, def dead %2, 12, implicit-def dead early-clobber $rax, 12, implicit-def dead early-clobber $rbx, 12, implicit-def dead early-clobber $rcx, 12, implicit-def dead early-clobber $rdx, 12, implicit-def dead early-clobber $rsi, 12, implicit-def dead early-clobber $rdi, 12, implicit-def dead early-clobber $rbp, 12, implicit-def dead early-clobber $r8, 12, implicit-def dead early-clobber $r9, 12, implicit-def dead early-clobber $r10, 12, implicit-def dead early-clobber $r11, 12, implicit-def dead early-clobber $r12, 12, implicit-def dead early-clobber $r13, 12, implicit-def dead early-clobber $r14, 12, implicit-def dead early-clobber $r15
  91     ; CHECK: $rdx = MOV64rm %stack.1, 1, $noreg, 0, $noreg :: (load 8 from %stack.1)
  92     ; CHECK: %3:gr64, dead %4:gr64 = MULX64rm %stack.0, 1, $noreg, 0, $noreg, implicit $rdx :: (load 8 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, 4063242, def dead %2, 12, implicit-def dead early-clobber $rax, 12, implicit-def dead early-clobber $rbx, 12, implicit-def dead early-clobber $rcx, 12, implicit-def dead early-clobber $rdx, 12, implicit-def dead early-clobber $rsi, 12, implicit-def dead early-clobber $rdi, 12, implicit-def dead early-clobber $rbp, 12, implicit-def dead early-clobber $r8, 12, implicit-def dead early-clobber $r9, 12, implicit-def dead early-clobber $r10, 12, implicit-def dead early-clobber $r11, 12, implicit-def dead early-clobber $r12, 12, implicit-def dead early-clobber $r13, 12, implicit-def dead early-clobber $r14, 12, 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 ...