58 lines · plain
1; Test memchr using SRST, with the correct prototype.2;3; RUN: llc < %s -mtriple=s390x-linux-gnu -no-integrated-as | FileCheck %s4 5declare ptr@memchr(ptr %src, i32 %char, i64 %len)6 7; Test a simple forwarded call.8define ptr@f1(i64 %len, ptr %src, i32 %char) {9; CHECK-LABEL: f1:10; CHECK-DAG: agr %r2, %r311; CHECK-DAG: llcr %r0, %r412; CHECK: [[LABEL:\.[^:]*]]:13; CHECK: srst %r2, %r314; CHECK-NEXT: jo [[LABEL]]15; CHECK: blr %r1416; CHECK: lghi %r2, 017; CHECK: br %r1418 %res = call ptr@memchr(ptr %src, i32 %char, i64 %len)19 ret ptr %res20}21 22; Test a doubled call with no use of %r0 in between. There should be a23; single load of %r0.24define ptr@f2(ptr %src, ptr %charptr, i64 %len) {25; CHECK-LABEL: f2:26; CHECK: llc %r0, 0(%r3)27; CHECK-NOT: %r028; CHECK: srst [[RES1:%r[1-5]]], %r229; CHECK-NOT: %r030; CHECK: srst %r2, [[RES1]]31; CHECK: br %r1432 %char = load volatile i8, ptr %charptr33 %charext = zext i8 %char to i3234 %res1 = call ptr@memchr(ptr %src, i32 %charext, i64 %len)35 %res2 = call ptr@memchr(ptr %res1, i32 %charext, i64 %len)36 ret ptr %res237}38 39; Test a doubled call with a use of %r0 in between. %r0 must be loaded40; for each loop.41define 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]]], %r246; CHECK: lhi %r0, 047; CHECK: blah %r048; CHECK: lr %r0, [[CHAR]]49; CHECK: srst %r2, [[RES1]]50; CHECK: br %r1451 %char = load volatile i8, ptr %charptr52 %charext = zext i8 %char to i3253 %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 %res257}58