test/CodeGen/AArch64/GlobalISel/regbank-fp-use-def.mir

   1 # NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
   2 # RUN: llc -O0 -mtriple arm64-- -run-pass=regbankselect %s -o - | FileCheck %s
   3
   4 # Check that we correctly assign register banks based off of instructions which
   5 # only use or only define FPRs.
   6 #
   7 # For example, G_SITOFP takes in a GPR, but only ever produces values on FPRs.
   8 # Some instructions can have inputs/outputs on either FPRs or GPRs. If one of
   9 # those instructions takes in the result of a G_SITOFP as a source, we should
  10 # put that source on a FPR.
  11 #
  12 # Similarly, G_FPTOSI can only take in a value on a FPR. So, if the result of
  13 # an instruction is consumed by a G_FPTOSI, we should put the instruction on
  14 # FPRs.
  15
  16 ---
  17 name:            load_only_uses_fp
  18 legalized:       true
  19 tracksRegLiveness: true
  20 body: |
  21   bb.0:
  22     liveins: $x0
  23     ; CHECK-LABEL: name: load_only_uses_fp
  24     ; CHECK: liveins: $x0
  25     ; CHECK: [[COPY:%[0-9]+]]:gpr(p0) = COPY $x0
  26     ; CHECK: [[C:%[0-9]+]]:fpr(s32) = G_FCONSTANT float 2.000000e+00
  27     ; CHECK: [[LOAD:%[0-9]+]]:fpr(s32) = G_LOAD [[COPY]](p0) :: (load 4)
  28     ; CHECK: [[FCMP:%[0-9]+]]:gpr(s32) = G_FCMP floatpred(uno), [[C]](s32), [[LOAD]]
  29     ; CHECK: $w0 = COPY [[FCMP]](s32)
  30     ; CHECK: RET_ReallyLR implicit $w0
  31     %0:_(p0) = COPY $x0
  32     %1:_(s32) = G_FCONSTANT float 2.0
  33     %2:_(s32) = G_LOAD %0 :: (load 4)
  34     %3:_(s32) = G_FCMP floatpred(uno), %1, %2
  35     $w0 = COPY %3(s32)
  36     RET_ReallyLR implicit $w0
  37 ...
  38 ---
  39 name:            unmerge_only_uses_fp
  40
  41 legalized:       true
  42 tracksRegLiveness: true
  43 body:             |
  44   bb.0:
  45     liveins: $x0
  46     ; CHECK-LABEL: name: unmerge_only_uses_fp
  47     ; CHECK: liveins: $x0
  48     ; CHECK: [[COPY:%[0-9]+]]:gpr(s64) = COPY $x0
  49     ; CHECK: [[COPY1:%[0-9]+]]:fpr(s64) = COPY [[COPY]](s64)
  50     ; CHECK: [[UV:%[0-9]+]]:fpr(s32), [[UV1:%[0-9]+]]:fpr(s32) = G_UNMERGE_VALUES [[COPY1]](s64)
  51     ; CHECK: [[FCMP:%[0-9]+]]:gpr(s32) = G_FCMP floatpred(uno), [[UV]](s32), [[UV1]]
  52     ; CHECK: $w0 = COPY [[FCMP]](s32)
  53     ; CHECK: RET_ReallyLR implicit $w0
  54     %0:_(s64) = COPY $x0
  55     %1:_(s32), %2:_(s32) = G_UNMERGE_VALUES %0(s64)
  56     %3:_(s32) = G_FCMP floatpred(uno), %1, %2
  57     $w0 = COPY %3(s32)
  58     RET_ReallyLR implicit $w0
  59
  60 ...
  61 ---
  62 name:            store_defined_by_fp
  63 legalized:       true
  64 tracksRegLiveness: true
  65 body: |
  66   bb.0:
  67     liveins: $x0, $w1
  68     ; CHECK-LABEL: name: store_defined_by_fp
  69     ; CHECK: liveins: $x0, $w1
  70     ; CHECK: [[COPY:%[0-9]+]]:gpr(p0) = COPY $x0
  71     ; CHECK: [[COPY1:%[0-9]+]]:gpr(s32) = COPY $w1
  72     ; CHECK: [[SITOFP:%[0-9]+]]:fpr(s32) = G_SITOFP [[COPY1]](s32)
  73     ; CHECK: G_STORE [[SITOFP]](s32), [[COPY]](p0) :: (store 4)
  74     %0:_(p0) = COPY $x0
  75     %1:_(s32) = COPY $w1
  76     %2:_(s32) = G_SITOFP %1
  77     G_STORE %2, %0 :: (store 4)
  78
  79 ...
  80 ---
  81 name:            select_defined_by_fp_using_fp
  82 legalized:       true
  83 tracksRegLiveness: true
  84 body:             |
  85   bb.0:
  86     liveins: $w0, $w1, $w2
  87     ; CHECK-LABEL: name: select_defined_by_fp_using_fp
  88     ; CHECK: liveins: $w0, $w1, $w2
  89     ; CHECK: [[COPY:%[0-9]+]]:gpr(s32) = COPY $w0
  90     ; CHECK: [[TRUNC:%[0-9]+]]:gpr(s1) = G_TRUNC %2(s32)
  91     ; CHECK: [[COPY1:%[0-9]+]]:gpr(s32) = COPY $w1
  92     ; CHECK: [[COPY2:%[0-9]+]]:gpr(s32) = COPY $w2
  93     ; CHECK: [[SITOFP:%[0-9]+]]:fpr(s32) = G_SITOFP [[COPY1]](s32)
  94     ; CHECK: [[COPY3:%[0-9]+]]:fpr(s32) = COPY [[COPY2]](s32)
  95     ; CHECK: [[SELECT:%[0-9]+]]:fpr(s32) = G_SELECT [[TRUNC]](s1), [[COPY3]], [[SITOFP]]
  96     ; CHECK: [[FPTOSI:%[0-9]+]]:gpr(s32) = G_FPTOSI [[SELECT]](s32)
  97     %0:_(s32) = COPY $w0
  98     %1:_(s1) = G_TRUNC %3(s32)
  99     %2:_(s32) = COPY $w1
 100     %3:_(s32) = COPY $w2
 101     %4:_(s32) = G_SITOFP %2
 102     %6:_(s32) = G_SELECT %1(s1), %3, %4
 103     %8:_(s32) = G_FPTOSI %6