test/CodeGen/AArch64/shift-mod.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -mtriple=aarch64 < %s | FileCheck %s
   3
   4 ; Check that we optimize out AND instructions and ADD/SUB instructions
   5 ; modulo the shift size to take advantage of the implicit mod done on
   6 ; the shift amount value by the variable shift/rotate instructions.
   7
   8 define i32 @test1(i32 %x, i64 %y) {
   9 ; CHECK-LABEL: test1:
  10 ; CHECK:       // %bb.0:
  11 ; CHECK-NEXT:    lsr w0, w0, w1
  12 ; CHECK-NEXT:    ret
  13   %sh_prom = trunc i64 %y to i32
  14   %shr = lshr i32 %x, %sh_prom
  15   ret i32 %shr
  16 }
  17
  18 define i64 @test2(i32 %x, i64 %y) {
  19 ; CHECK-LABEL: test2:
  20 ; CHECK:       // %bb.0:
  21 ; CHECK-NEXT:    neg w[[REG:[0-9]+]], w0
  22 ; CHECK-NEXT:    asr x0, x1, x[[REG]]
  23 ; CHECK-NEXT:    ret
  24   %sub9 = sub nsw i32 64, %x
  25   %sh_prom12.i = zext i32 %sub9 to i64
  26   %shr.i = ashr i64 %y, %sh_prom12.i
  27   ret i64 %shr.i
  28 }
  29
  30 define i64 @test3(i64 %x, i64 %y) {
  31 ; CHECK-LABEL: test3:
  32 ; CHECK:       // %bb.0:
  33 ; CHECK-NEXT:    lsl x0, x1, x0
  34 ; CHECK-NEXT:    ret
  35   %add = add nsw i64 64, %x
  36   %shl = shl i64 %y, %add
  37   ret i64 %shl
  38 }
  39
  40 define i64 @test4(i64 %y, i32 %s) {
  41 ; CHECK-LABEL: test4:
  42 ; CHECK:       // %bb.0: // %entry
  43 ; CHECK-NEXT:    // kill: def $w1 killed $w1 def $x1
  44 ; CHECK-NEXT:    asr x0, x0, x1
  45 ; CHECK-NEXT:    ret
  46 entry:
  47   %sh_prom = zext i32 %s to i64
  48   %shr = ashr i64 %y, %sh_prom
  49   ret i64 %shr
  50 }
  51
  52 define i64 @test5(i64 %y, i32 %s) {
  53 ; CHECK-LABEL: test5:
  54 ; CHECK:       // %bb.0: // %entry
  55 ; CHECK-NEXT:    // kill: def $w1 killed $w1 def $x1
  56 ; CHECK-NEXT:    asr x0, x0, x1
  57 ; CHECK-NEXT:    ret
  58 entry:
  59   %sh_prom = sext i32 %s to i64
  60   %shr = ashr i64 %y, %sh_prom
  61   ret i64 %shr
  62 }
  63
  64 define i64 @test6(i64 %y, i32 %s) {
  65 ; CHECK-LABEL: test6:
  66 ; CHECK:       // %bb.0: // %entry
  67 ; CHECK-NEXT:    // kill: def $w1 killed $w1 def $x1
  68 ; CHECK-NEXT:    lsl x0, x0, x1
  69 ; CHECK-NEXT:    ret
  70 entry:
  71   %sh_prom = sext i32 %s to i64
  72   %shr = shl i64 %y, %sh_prom
  73   ret i64 %shr
  74 }