llvm/test/CodeGen/LoongArch/ir-instruction/lshr.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc --mtriple=loongarch32 -mattr=+d < %s | FileCheck %s --check-prefix=LA32
   3 ; RUN: llc --mtriple=loongarch64 -mattr=+d < %s | FileCheck %s --check-prefix=LA64
   4
   5 ;; Exercise the 'lshr' LLVM IR: https://llvm.org/docs/LangRef.html#lshr-instruction
   6
   7 define i1 @lshr_i1(i1 %x, i1 %y) {
   8 ; LA32-LABEL: lshr_i1:
   9 ; LA32:       # %bb.0:
  10 ; LA32-NEXT:    ret
  11 ;
  12 ; LA64-LABEL: lshr_i1:
  13 ; LA64:       # %bb.0:
  14 ; LA64-NEXT:    ret
  15   %lshr = lshr i1 %x, %y
  16   ret i1 %lshr
  17 }
  18
  19 define i8 @lshr_i8(i8 %x, i8 %y) {
  20 ; LA32-LABEL: lshr_i8:
  21 ; LA32:       # %bb.0:
  22 ; LA32-NEXT:    andi $a0, $a0, 255
  23 ; LA32-NEXT:    srl.w $a0, $a0, $a1
  24 ; LA32-NEXT:    ret
  25 ;
  26 ; LA64-LABEL: lshr_i8:
  27 ; LA64:       # %bb.0:
  28 ; LA64-NEXT:    andi $a0, $a0, 255
  29 ; LA64-NEXT:    srl.d $a0, $a0, $a1
  30 ; LA64-NEXT:    ret
  31   %lshr = lshr i8 %x, %y
  32   ret i8 %lshr
  33 }
  34
  35 define i16 @lshr_i16(i16 %x, i16 %y) {
  36 ; LA32-LABEL: lshr_i16:
  37 ; LA32:       # %bb.0:
  38 ; LA32-NEXT:    bstrpick.w $a0, $a0, 15, 0
  39 ; LA32-NEXT:    srl.w $a0, $a0, $a1
  40 ; LA32-NEXT:    ret
  41 ;
  42 ; LA64-LABEL: lshr_i16:
  43 ; LA64:       # %bb.0:
  44 ; LA64-NEXT:    bstrpick.d $a0, $a0, 15, 0
  45 ; LA64-NEXT:    srl.d $a0, $a0, $a1
  46 ; LA64-NEXT:    ret
  47   %lshr = lshr i16 %x, %y
  48   ret i16 %lshr
  49 }
  50
  51 define i32 @lshr_i32(i32 %x, i32 %y) {
  52 ; LA32-LABEL: lshr_i32:
  53 ; LA32:       # %bb.0:
  54 ; LA32-NEXT:    srl.w $a0, $a0, $a1
  55 ; LA32-NEXT:    ret
  56 ;
  57 ; LA64-LABEL: lshr_i32:
  58 ; LA64:       # %bb.0:
  59 ; LA64-NEXT:    srl.w $a0, $a0, $a1
  60 ; LA64-NEXT:    ret
  61   %lshr = lshr i32 %x, %y
  62   ret i32 %lshr
  63 }
  64
  65 define i64 @lshr_i64(i64 %x, i64 %y) {
  66 ; LA32-LABEL: lshr_i64:
  67 ; LA32:       # %bb.0:
  68 ; LA32-NEXT:    srl.w $a0, $a0, $a2
  69 ; LA32-NEXT:    xori $a3, $a2, 31
  70 ; LA32-NEXT:    slli.w $a4, $a1, 1
  71 ; LA32-NEXT:    sll.w $a3, $a4, $a3
  72 ; LA32-NEXT:    or $a0, $a0, $a3
  73 ; LA32-NEXT:    addi.w $a3, $a2, -32
  74 ; LA32-NEXT:    slti $a4, $a3, 0
  75 ; LA32-NEXT:    maskeqz $a0, $a0, $a4
  76 ; LA32-NEXT:    srl.w $a5, $a1, $a3
  77 ; LA32-NEXT:    masknez $a4, $a5, $a4
  78 ; LA32-NEXT:    or $a0, $a0, $a4
  79 ; LA32-NEXT:    srl.w $a1, $a1, $a2
  80 ; LA32-NEXT:    srai.w $a2, $a3, 31
  81 ; LA32-NEXT:    and $a1, $a2, $a1
  82 ; LA32-NEXT:    ret
  83 ;
  84 ; LA64-LABEL: lshr_i64:
  85 ; LA64:       # %bb.0:
  86 ; LA64-NEXT:    srl.d $a0, $a0, $a1
  87 ; LA64-NEXT:    ret
  88   %lshr = lshr i64 %x, %y
  89   ret i64 %lshr
  90 }
  91
  92 define i1 @lshr_i1_3(i1 %x) {
  93 ; LA32-LABEL: lshr_i1_3:
  94 ; LA32:       # %bb.0:
  95 ; LA32-NEXT:    ret
  96 ;
  97 ; LA64-LABEL: lshr_i1_3:
  98 ; LA64:       # %bb.0:
  99 ; LA64-NEXT:    ret
 100   %lshr = lshr i1 %x, 3
 101   ret i1 %lshr
 102 }
 103
 104 define i8 @lshr_i8_3(i8 %x) {
 105 ; LA32-LABEL: lshr_i8_3:
 106 ; LA32:       # %bb.0:
 107 ; LA32-NEXT:    bstrpick.w $a0, $a0, 7, 3
 108 ; LA32-NEXT:    ret
 109 ;
 110 ; LA64-LABEL: lshr_i8_3:
 111 ; LA64:       # %bb.0:
 112 ; LA64-NEXT:    bstrpick.d $a0, $a0, 7, 3
 113 ; LA64-NEXT:    ret
 114   %lshr = lshr i8 %x, 3
 115   ret i8 %lshr
 116 }
 117
 118 define i16 @lshr_i16_3(i16 %x) {
 119 ; LA32-LABEL: lshr_i16_3:
 120 ; LA32:       # %bb.0:
 121 ; LA32-NEXT:    bstrpick.w $a0, $a0, 15, 3
 122 ; LA32-NEXT:    ret
 123 ;
 124 ; LA64-LABEL: lshr_i16_3:
 125 ; LA64:       # %bb.0:
 126 ; LA64-NEXT:    bstrpick.d $a0, $a0, 15, 3
 127 ; LA64-NEXT:    ret
 128   %lshr = lshr i16 %x, 3
 129   ret i16 %lshr
 130 }
 131
 132 define i32 @lshr_i32_3(i32 %x) {
 133 ; LA32-LABEL: lshr_i32_3:
 134 ; LA32:       # %bb.0:
 135 ; LA32-NEXT:    srli.w $a0, $a0, 3
 136 ; LA32-NEXT:    ret
 137 ;
 138 ; LA64-LABEL: lshr_i32_3:
 139 ; LA64:       # %bb.0:
 140 ; LA64-NEXT:    bstrpick.d $a0, $a0, 31, 3
 141 ; LA64-NEXT:    ret
 142   %lshr = lshr i32 %x, 3
 143   ret i32 %lshr
 144 }
 145
 146 define i64 @lshr_i64_3(i64 %x) {
 147 ; LA32-LABEL: lshr_i64_3:
 148 ; LA32:       # %bb.0:
 149 ; LA32-NEXT:    slli.w $a2, $a1, 29
 150 ; LA32-NEXT:    srli.w $a0, $a0, 3
 151 ; LA32-NEXT:    or $a0, $a0, $a2
 152 ; LA32-NEXT:    srli.w $a1, $a1, 3
 153 ; LA32-NEXT:    ret
 154 ;
 155 ; LA64-LABEL: lshr_i64_3:
 156 ; LA64:       # %bb.0:
 157 ; LA64-NEXT:    srli.d $a0, $a0, 3
 158 ; LA64-NEXT:    ret
 159   %lshr = lshr i64 %x, 3
 160   ret i64 %lshr
 161 }