test/CodeGen/AArch64/i128-shift.ll

   1 ; RUN: llc -verify-machineinstrs < %s -mtriple=aarch64-none-linux-gnu | FileCheck %s
   2 ; arm64 has its own version of this in long-shift.ll. We'll just use that.
   3
   4 define i128 @test_i128_lsl(i128 %a, i32 %shift) {
   5 ; CHECK-LABEL: test_i128_lsl:
   6
   7   %sh_prom = zext i32 %shift to i128
   8   %shl = shl i128 %a, %sh_prom
   9
  10 ; CHECK: movz [[SIXTYFOUR:x[0-9]+]], #64
  11 ; CHECK-NEXT: sub [[REVSHAMT:x[0-9]+]], [[SIXTYFOUR]], [[SHAMT_32:w[0-9]+]], uxtw
  12 ; CHECK-NEXT: lsr [[TMP1:x[0-9]+]], [[LO:x[0-9]+]], [[REVSHAMT]]
  13 ; CHECK: lsl [[TMP2:x[0-9]+]], [[HI:x[0-9]+]], [[SHAMT:x[0-9]+]]
  14 ; CHECK-NEXT: orr [[FALSEVAL:x[0-9]+]], [[TMP1]], [[TMP2]]
  15 ; CHECK-NEXT: sub [[EXTRASHAMT:x[0-9]+]], [[SHAMT]], #64
  16 ; CHECK-NEXT: lsl [[TMP3:x[0-9]+]], [[LO]], [[EXTRASHAMT]]
  17 ; CHECK-NEXT: cmp [[EXTRASHAMT]], #0
  18 ; CHECK-NEXT: csel [[RESULTHI:x[0-9]+]], [[TMP3]], [[FALSEVAL]], ge
  19 ; CHECK-NEXT: lsl [[TMP4:x[0-9]+]], [[LO]], [[SHAMT]]
  20 ; CHECK-NEXT: csel [[RESULTLO:x[0-9]+]], xzr, [[TMP4]], ge
  21
  22   ret i128 %shl
  23 }
  24
  25 define i128 @test_i128_shr(i128 %a, i32 %shift) {
  26 ; CHECK-LABEL: test_i128_shr:
  27
  28   %sh_prom = zext i32 %shift to i128
  29   %shr = lshr i128 %a, %sh_prom
  30
  31 ; CHECK: movz [[SIXTYFOUR]], #64
  32 ; CHECK-NEXT: sub [[REVSHAMT:x[0-9]+]], [[SIXTYFOUR]], [[SHAMT_32:w[0-9]+]], uxtw
  33 ; CHECK-NEXT: lsl [[TMP2:x[0-9]+]], [[HI:x[0-9]+]], [[REVSHAMT]]
  34 ; CHECK: lsr [[TMP1:x[0-9]+]], [[LO:x[0-9]+]], [[SHAMT:x[0-9]+]]
  35 ; CHECK-NEXT: orr [[FALSEVAL:x[0-9]+]], [[TMP1]], [[TMP2]]
  36 ; CHECK-NEXT: sub [[EXTRASHAMT:x[0-9]+]], [[SHAMT]], #64
  37 ; CHECK-NEXT: lsr [[TRUEVAL:x[0-9]+]], [[HI]], [[EXTRASHAMT]]
  38 ; CHECK-NEXT: cmp [[EXTRASHAMT]], #0
  39 ; CHECK-NEXT: csel [[RESULTLO:x[0-9]+]], [[TRUEVAL]], [[FALSEVAL]], ge
  40 ; CHECK-NEXT: lsr [[TMP3:x[0-9]+]], [[HI]], [[SHAMT]]
  41 ; CHECK-NEXT: csel [[RESULTHI:x[0-9]+]], xzr, [[TMP3]], ge
  42
  43   ret i128 %shr
  44 }