llvm/test/CodeGen/AArch64/GlobalISel/form-bitfield-extract-from-sextinreg.mir

   1 # NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
   2 # RUN: llc -mtriple aarch64 -run-pass=aarch64-postlegalizer-combiner -verify-machineinstrs %s -o - | FileCheck %s
   3
   4 # Check that we can fold a G_SEXT_INREG fed by a G_ASHR/G_LSHR into a G_SBFX.
   5
   6 ...
   7 ---
   8 name:            sextinreg_ashr_to_sbfx
   9 tracksRegLiveness: true
  10 legalized: true
  11 body:             |
  12   bb.0:
  13     liveins: $w0
  14     ; CHECK-LABEL: name: sextinreg_ashr_to_sbfx
  15     ; CHECK: liveins: $w0
  16     ; CHECK-NEXT: {{  $}}
  17     ; CHECK-NEXT: %x:_(s32) = COPY $w0
  18     ; CHECK-NEXT: %lsb:_(s32) = G_CONSTANT i32 5
  19     ; CHECK-NEXT: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 10
  20     ; CHECK-NEXT: %sext_inreg:_(s32) = G_SBFX %x, %lsb(s32), [[C]]
  21     ; CHECK-NEXT: $w0 = COPY %sext_inreg(s32)
  22     ; CHECK-NEXT: RET_ReallyLR implicit $w0
  23     %x:_(s32) = COPY $w0
  24     %lsb:_(s32) = G_CONSTANT i32 5
  25     %shift:_(s32) = G_ASHR %x, %lsb
  26     %sext_inreg:_(s32) = G_SEXT_INREG %shift, 10
  27     $w0 = COPY %sext_inreg
  28     RET_ReallyLR implicit $w0
  29
  30 ...
  31 ---
  32 name:            sextinreg_lshr_to_sbfx
  33 tracksRegLiveness: true
  34 legalized: true
  35 body:             |
  36   bb.0:
  37     liveins: $w0
  38     ; CHECK-LABEL: name: sextinreg_lshr_to_sbfx
  39     ; CHECK: liveins: $w0
  40     ; CHECK-NEXT: {{  $}}
  41     ; CHECK-NEXT: %x:_(s32) = COPY $w0
  42     ; CHECK-NEXT: %lsb:_(s32) = G_CONSTANT i32 5
  43     ; CHECK-NEXT: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 10
  44     ; CHECK-NEXT: %sext_inreg:_(s32) = G_SBFX %x, %lsb(s32), [[C]]
  45     ; CHECK-NEXT: $w0 = COPY %sext_inreg(s32)
  46     ; CHECK-NEXT: RET_ReallyLR implicit $w0
  47     %x:_(s32) = COPY $w0
  48     %lsb:_(s32) = G_CONSTANT i32 5
  49     %shift:_(s32) = G_LSHR %x, %lsb
  50     %sext_inreg:_(s32) = G_SEXT_INREG %shift, 10
  51     $w0 = COPY %sext_inreg
  52     RET_ReallyLR implicit $w0
  53
  54
  55 ...
  56 ---
  57 name:            dont_apply_no_constant
  58 tracksRegLiveness: true
  59 legalized: true
  60 body:             |
  61   bb.0:
  62     liveins: $w0
  63     ; AArch64 needs a constant on the shift for this combine.
  64
  65     ; CHECK-LABEL: name: dont_apply_no_constant
  66     ; CHECK: liveins: $w0
  67     ; CHECK-NEXT: {{  $}}
  68     ; CHECK-NEXT: %x:_(s32) = COPY $w0
  69     ; CHECK-NEXT: %y:_(s32) = COPY $w0
  70     ; CHECK-NEXT: %shift:_(s32) = G_LSHR %x, %y(s32)
  71     ; CHECK-NEXT: %sext_inreg:_(s32) = G_SEXT_INREG %shift, 10
  72     ; CHECK-NEXT: $w0 = COPY %sext_inreg(s32)
  73     ; CHECK-NEXT: RET_ReallyLR implicit $w0
  74     %x:_(s32) = COPY $w0
  75     %y:_(s32) = COPY $w0
  76     %shift:_(s32) = G_LSHR %x, %y
  77     %sext_inreg:_(s32) = G_SEXT_INREG %shift, 10
  78     $w0 = COPY %sext_inreg
  79     RET_ReallyLR implicit $w0
  80
  81 ...
  82 ---
  83 name:            dont_apply_shift_imm_too_large
  84 tracksRegLiveness: true
  85 legalized: true
  86 body:             |
  87   bb.0:
  88     liveins: $w0
  89
  90     ; LSB must be in 0-31.
  91
  92     ; CHECK-LABEL: name: dont_apply_shift_imm_too_large
  93     ; CHECK: liveins: $w0
  94     ; CHECK-NEXT: {{  $}}
  95     ; CHECK-NEXT: %x:_(s32) = COPY $w0
  96     ; CHECK-NEXT: %lsb:_(s32) = G_CONSTANT i32 32
  97     ; CHECK-NEXT: %shift:_(s32) = G_ASHR %x, %lsb(s32)
  98     ; CHECK-NEXT: $w0 = COPY %shift(s32)
  99     ; CHECK-NEXT: RET_ReallyLR implicit $w0
 100     %x:_(s32) = COPY $w0
 101     %lsb:_(s32) = G_CONSTANT i32 32
 102     %shift:_(s32) = G_ASHR %x, %lsb
 103     %sext_inreg:_(s32) = G_SEXT_INREG %shift, 1
 104     $w0 = COPY %sext_inreg
 105     RET_ReallyLR implicit $w0
 106
 107 ...
 108 ---
 109 name:            dont_apply_negative_shift_imm
 110 tracksRegLiveness: true
 111 legalized: true
 112 body:             |
 113   bb.0:
 114     liveins: $w0
 115
 116     ; LSB must be in 0-31.
 117
 118     ; CHECK-LABEL: name: dont_apply_negative_shift_imm
 119     ; CHECK: liveins: $w0
 120     ; CHECK-NEXT: {{  $}}
 121     ; CHECK-NEXT: %x:_(s32) = COPY $w0
 122     ; CHECK-NEXT: %lsb:_(s32) = G_CONSTANT i32 -1
 123     ; CHECK-NEXT: %shift:_(s32) = G_ASHR %x, %lsb(s32)
 124     ; CHECK-NEXT: $w0 = COPY %shift(s32)
 125     ; CHECK-NEXT: RET_ReallyLR implicit $w0
 126     %x:_(s32) = COPY $w0
 127     %lsb:_(s32) = G_CONSTANT i32 -1
 128     %shift:_(s32) = G_ASHR %x, %lsb
 129     %sext_inreg:_(s32) = G_SEXT_INREG %shift, 1
 130     $w0 = COPY %sext_inreg
 131     RET_ReallyLR implicit $w0
 132
 133 ...
 134 ---
 135 name:            dont_apply_more_than_one_use
 136 tracksRegLiveness: true
 137 legalized: true
 138 body:             |
 139   bb.0:
 140     liveins: $w0
 141     ; CHECK-LABEL: name: dont_apply_more_than_one_use
 142     ; CHECK: liveins: $w0
 143     ; CHECK-NEXT: {{  $}}
 144     ; CHECK-NEXT: %x:_(s32) = COPY $w0
 145     ; CHECK-NEXT: %lsb:_(s32) = G_CONSTANT i32 1
 146     ; CHECK-NEXT: %shift:_(s32) = G_ASHR %x, %lsb(s32)
 147     ; CHECK-NEXT: %sext_inreg:_(s32) = G_SEXT_INREG %shift, 1
 148     ; CHECK-NEXT: %mul:_(s32) = G_MUL %shift, %sext_inreg
 149     ; CHECK-NEXT: $w0 = COPY %mul(s32)
 150     ; CHECK-NEXT: RET_ReallyLR implicit $w0
 151     %x:_(s32) = COPY $w0
 152     %lsb:_(s32) = G_CONSTANT i32 1
 153     %shift:_(s32) = G_ASHR %x, %lsb
 154     %sext_inreg:_(s32) = G_SEXT_INREG %shift, 1
 155     %mul:_(s32) = G_MUL %shift, %sext_inreg
 156     $w0 = COPY %mul
 157     RET_ReallyLR implicit $w0