test/CodeGen/AArch64/GlobalISel/legalize-extracts.mir

   1 # NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
   2 # RUN: llc -mtriple=aarch64-linux-gnu -O0 -run-pass=legalizer %s -o - | FileCheck %s
   3
   4 ---
   5 name:            test_extracts_1
   6 body: |
   7   bb.0:
   8     liveins: $w0
   9
  10     ; Low part of extraction takes entirity of the low register entirely, so
  11     ; value stored is forwarded directly from first load.
  12
  13     ; CHECK-LABEL: name: test_extracts_1
  14     ; CHECK: [[COPY:%[0-9]+]]:_(s64) = COPY $x0
  15     ; CHECK: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
  16     ; CHECK: [[COPY2:%[0-9]+]]:_(p0) = COPY $x2
  17     ; CHECK: [[LOAD:%[0-9]+]]:_(s64) = G_LOAD [[COPY2]](p0) :: (load 8, align 16)
  18     ; CHECK: [[C:%[0-9]+]]:_(s64) = G_CONSTANT i64 8
  19     ; CHECK: [[GEP:%[0-9]+]]:_(p0) = G_GEP [[COPY2]], [[C]](s64)
  20     ; CHECK: [[LOAD1:%[0-9]+]]:_(s64) = G_LOAD [[GEP]](p0) :: (load 8)
  21     ; CHECK: [[COPY3:%[0-9]+]]:_(s64) = COPY [[LOAD]](s64)
  22     ; CHECK: G_STORE [[COPY3]](s64), [[COPY2]](p0) :: (store 8)
  23     ; CHECK: RET_ReallyLR
  24     %0:_(s64) = COPY $x0
  25     %1:_(s32) = COPY $w1
  26     %2:_(p0) = COPY $x2
  27     %3:_(s128) = G_LOAD %2(p0) :: (load 16)
  28     %4:_(s64) = G_EXTRACT %3(s128), 0
  29     G_STORE %4(s64), %2(p0) :: (store 8)
  30     RET_ReallyLR
  31 ...
  32
  33 ---
  34 name:            test_extracts_2
  35 body: |
  36   bb.0:
  37     liveins: $w0
  38
  39       ; Low extraction wipes takes whole low register. High extraction is real.
  40     ; CHECK-LABEL: name: test_extracts_2
  41     ; CHECK: [[COPY:%[0-9]+]]:_(s64) = COPY $x0
  42     ; CHECK: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
  43     ; CHECK: [[COPY2:%[0-9]+]]:_(p0) = COPY $x2
  44     ; CHECK: [[LOAD:%[0-9]+]]:_(s64) = G_LOAD [[COPY2]](p0) :: (load 8, align 16)
  45     ; CHECK: [[C:%[0-9]+]]:_(s64) = G_CONSTANT i64 8
  46     ; CHECK: [[GEP:%[0-9]+]]:_(p0) = G_GEP [[COPY2]], [[C]](s64)
  47     ; CHECK: [[LOAD1:%[0-9]+]]:_(s64) = G_LOAD [[GEP]](p0) :: (load 8)
  48     ; CHECK: [[COPY3:%[0-9]+]]:_(s64) = COPY [[LOAD]](s64)
  49     ; CHECK: [[EXTRACT:%[0-9]+]]:_(s32) = G_EXTRACT [[LOAD1]](s64), 0
  50     ; CHECK: G_STORE [[COPY3]](s64), [[COPY2]](p0) :: (store 8)
  51     ; CHECK: G_STORE [[EXTRACT]](s32), [[COPY2]](p0) :: (store 4)
  52     ; CHECK: RET_ReallyLR
  53     %0:_(s64) = COPY $x0
  54     %1:_(s32) = COPY $w1
  55     %2:_(p0) = COPY $x2
  56     %3:_(s128) = G_LOAD %2(p0) :: (load 16)
  57     %4:_(s64) = G_EXTRACT %3(s128), 0
  58     %5:_(s32) = G_EXTRACT %3(s128), 64
  59     G_STORE %4(s64), %2(p0) :: (store 8)
  60     G_STORE %5(s32), %2(p0) :: (store 4)
  61     RET_ReallyLR
  62 ...
  63
  64 ---
  65 name:            test_extracts_3
  66 body: |
  67   bb.0:
  68     liveins: $x0, $x1, $x2
  69
  70
  71     ; CHECK-LABEL: name: test_extracts_3
  72     ; CHECK: [[COPY:%[0-9]+]]:_(s64) = COPY $x0
  73     ; CHECK: [[COPY1:%[0-9]+]]:_(s64) = COPY $x1
  74     ; CHECK: [[EXTRACT:%[0-9]+]]:_(s32) = G_EXTRACT [[COPY]](s64), 32
  75     ; CHECK: [[EXTRACT1:%[0-9]+]]:_(s32) = G_EXTRACT [[COPY1]](s64), 0
  76     ; CHECK: [[MV:%[0-9]+]]:_(s64) = G_MERGE_VALUES [[EXTRACT]](s32), [[EXTRACT1]](s32)
  77     ; CHECK: $x0 = COPY [[MV]](s64)
  78     ; CHECK: RET_ReallyLR
  79     %0:_(s64) = COPY $x0
  80     %1:_(s64) = COPY $x1
  81     %2:_(s128) = G_MERGE_VALUES %0, %1
  82     %3:_(s64) = G_EXTRACT %2, 32
  83     $x0 = COPY %3
  84     RET_ReallyLR
  85 ...
  86
  87 ---
  88 name:            test_extracts_4
  89 body: |
  90   bb.0:
  91     liveins: $x0, $x1, $x2
  92
  93
  94     ; CHECK-LABEL: name: test_extracts_4
  95     ; CHECK: [[COPY:%[0-9]+]]:_(s64) = COPY $x0
  96     ; CHECK: [[COPY1:%[0-9]+]]:_(s64) = COPY $x1
  97     ; CHECK: [[EXTRACT:%[0-9]+]]:_(s32) = G_EXTRACT [[COPY]](s64), 32
  98     ; CHECK: $w0 = COPY [[EXTRACT]](s32)
  99     ; CHECK: RET_ReallyLR
 100     %0:_(s64) = COPY $x0
 101     %1:_(s64) = COPY $x1
 102     %2:_(s128) = G_MERGE_VALUES %0, %1
 103     %3:_(s32) = G_EXTRACT %2, 32
 104     $w0 = COPY %3
 105     RET_ReallyLR
 106 ...