test/CodeGen/BPF/32-bit-subreg-peephole.ll

   1 ; RUN: llc -O2 -march=bpfel -mcpu=v2 -mattr=+alu32 < %s | FileCheck %s
   2 ;
   3 ; long long select_u(unsigned a, unsigned b, long long c, long long d)
   4 ; {
   5 ;   if (a > b)
   6 ;     return c;
   7 ;   else
   8 ;     return d;
   9 ; }
  10 ;
  11 ; long long select_u_2(unsigned a, unsigned long long b, long long c, long long d)
  12 ; {
  13 ;   if (a > b)
  14 ;     return c;
  15 ;   else
  16 ;     return d;
  17 ; }
  18 ;
  19 ; long long select_s(signed a, signed b, long long c, long long d)
  20 ; {
  21 ;   if (a > b)
  22 ;     return c;
  23 ;   else
  24 ;     return d;
  25 ; }
  26 ;
  27 ; long long bar ();
  28 ;
  29 ; int foo (int b, int c)
  30 ; {
  31 ;   unsigned int i32_val = (unsigned int) bar();
  32 ;
  33 ;   if (i32_val < 10)
  34 ;     return b;
  35 ;   else
  36 ;     return c;
  37 ; }
  38 ;
  39 ; int *inc_p (int *p, unsigned a)
  40 ; {
  41 ;   return p + a;
  42 ; }
  43
  44 ; Function Attrs: norecurse nounwind readnone
  45 define dso_local i64 @select_u(i32 %a, i32 %b, i64 %c, i64 %d) local_unnamed_addr #0 {
  46 ; CHECK-LABEL: select_u:
  47 entry:
  48   %cmp = icmp ugt i32 %a, %b
  49   %c.d = select i1 %cmp, i64 %c, i64 %d
  50 ; CHECK-NOT: r{{[0-9]+}} <<= 32
  51 ; CHECK-NOT: r{{[0-9]+}} >>= 32
  52 ; CHECK: if r{{[0-9]+}} {{<|>}} r{{[0-9]+}} goto
  53   ret i64 %c.d
  54 }
  55
  56 ; Function Attrs: norecurse nounwind readnone
  57 define dso_local i64 @select_u_2(i32 %a, i64 %b, i64 %c, i64 %d) local_unnamed_addr #0 {
  58 ; CHECK-LABEL: select_u_2:
  59 entry:
  60   %conv = zext i32 %a to i64
  61 ; CHECK-NOT: r{{[0-9]+}} <<= 32
  62 ; CHECK-NOT: r{{[0-9]+}} >>= 32
  63   %cmp = icmp ugt i64 %conv, %b
  64   %c.d = select i1 %cmp, i64 %c, i64 %d
  65   ret i64 %c.d
  66 }
  67
  68 ; Function Attrs: norecurse nounwind readnone
  69 define dso_local i64 @select_s(i32 %a, i32 %b, i64 %c, i64 %d) local_unnamed_addr #0 {
  70 ; CHECK-LABEL: select_s:
  71 entry:
  72   %cmp = icmp sgt i32 %a, %b
  73   %c.d = select i1 %cmp, i64 %c, i64 %d
  74 ; CHECK: r{{[0-9]+}} <<= 32
  75 ; CHECK-NEXT: r{{[0-9]+}} s>>= 32
  76 ; CHECK: if r{{[0-9]+}} s{{<|>}} r{{[0-9]+}} goto
  77   ret i64 %c.d
  78 }
  79
  80 ; Function Attrs: nounwind
  81 define dso_local i32 @foo(i32 %b, i32 %c) local_unnamed_addr #0 {
  82 ; CHECK-LABEL: foo:
  83 entry:
  84   %call = tail call i64 bitcast (i64 (...)* @bar to i64 ()*)() #2
  85   %conv = trunc i64 %call to i32
  86   %cmp = icmp ult i32 %conv, 10
  87 ; The shifts can't be optimized out because %call comes from function call
  88 ; returning i64 so the high bits might be valid.
  89 ; CHECK: r{{[0-9]+}} <<= 32
  90 ; CHECK-NEXT: r{{[0-9]+}} >>= 32
  91   %b.c = select i1 %cmp, i32 %b, i32 %c
  92 ; CHECK: if r{{[0-9]+}} {{<|>}} {{[0-9]+}} goto
  93   ret i32 %b.c
  94 }
  95
  96 declare dso_local i64 @bar(...) local_unnamed_addr #1
  97
  98 ; Function Attrs: norecurse nounwind readnone
  99 define dso_local i32* @inc_p(i32* readnone %p, i32 %a) local_unnamed_addr #0 {
 100 ; CHECK-LABEL: inc_p:
 101 entry:
 102   %idx.ext = zext i32 %a to i64
 103 ; CHECK-NOT: r{{[0-9]+}} <<= 32
 104 ; CHECK-NOT: r{{[0-9]+}} >>= 32
 105   %add.ptr = getelementptr inbounds i32, i32* %p, i64 %idx.ext
 106   ret i32* %add.ptr
 107 }