llvm/test/CodeGen/SystemZ/memchr-01.ll

   1 ; Test memchr using SRST, with the correct prototype.
   2 ;
   3 ; RUN: llc < %s -mtriple=s390x-linux-gnu -no-integrated-as | FileCheck %s
   4
   5 declare ptr@memchr(ptr %src, i32 %char, i64 %len)
   6
   7 ; Test a simple forwarded call.
   8 define ptr@f1(i64 %len, ptr %src, i32 %char) {
   9 ; CHECK-LABEL: f1:
  10 ; CHECK-DAG: agr %r2, %r3
  11 ; CHECK-DAG: llcr %r0, %r4
  12 ; CHECK: [[LABEL:\.[^:]*]]:
  13 ; CHECK: srst %r2, %r3
  14 ; CHECK-NEXT: jo [[LABEL]]
  15 ; CHECK: blr %r14
  16 ; CHECK: lghi %r2, 0
  17 ; CHECK: br %r14
  18   %res = call ptr@memchr(ptr %src, i32 %char, i64 %len)
  19   ret ptr %res
  20 }
  21
  22 ; Test a doubled call with no use of %r0 in between.  There should be a
  23 ; single load of %r0.
  24 define ptr@f2(ptr %src, ptr %charptr, i64 %len) {
  25 ; CHECK-LABEL: f2:
  26 ; CHECK: llc %r0, 0(%r3)
  27 ; CHECK-NOT: %r0
  28 ; CHECK: srst [[RES1:%r[1-5]]], %r2
  29 ; CHECK-NOT: %r0
  30 ; CHECK: srst %r2, [[RES1]]
  31 ; CHECK: br %r14
  32   %char = load volatile i8, ptr %charptr
  33   %charext = zext i8 %char to i32
  34   %res1 = call ptr@memchr(ptr %src, i32 %charext, i64 %len)
  35   %res2 = call ptr@memchr(ptr %res1, i32 %charext, i64 %len)
  36   ret ptr %res2
  37 }
  38
  39 ; Test a doubled call with a use of %r0 in between.  %r0 must be loaded
  40 ; for each loop.
  41 define ptr@f3(ptr %src, ptr %charptr, i64 %len) {
  42 ; CHECK-LABEL: f3:
  43 ; CHECK: llc [[CHAR:%r[1-5]]], 0(%r3)
  44 ; CHECK: lr %r0, [[CHAR]]
  45 ; CHECK: srst [[RES1:%r[1-5]]], %r2
  46 ; CHECK: lhi %r0, 0
  47 ; CHECK: blah %r0
  48 ; CHECK: lr %r0, [[CHAR]]
  49 ; CHECK: srst %r2, [[RES1]]
  50 ; CHECK: br %r14
  51   %char = load volatile i8, ptr %charptr
  52   %charext = zext i8 %char to i32
  53   %res1 = call ptr@memchr(ptr %src, i32 %charext, i64 %len)
  54   call void asm sideeffect "blah $0", "{r0}" (i32 0)
  55   %res2 = call ptr@memchr(ptr %res1, i32 %charext, i64 %len)
  56   ret ptr %res2
  57 }