125 lines · plain
1; Test STOCGs that are presented as selects.2;3; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z196 | FileCheck %s4 5declare void @foo(ptr)6 7; Test with the loaded value first.8define void @f1(ptr %ptr, i64 %alt, i32 %limit) {9; CHECK-LABEL: f1:10; CHECK: clfi %r4, 4211; CHECK: stocghe %r3, 0(%r2)12; CHECK: br %r1413 %cond = icmp ult i32 %limit, 4214 %orig = load i64, ptr %ptr15 %res = select i1 %cond, i64 %orig, i64 %alt16 store i64 %res, ptr %ptr17 ret void18}19 20; ...and with the loaded value second21define void @f2(ptr %ptr, i64 %alt, i32 %limit) {22; CHECK-LABEL: f2:23; CHECK: clfi %r4, 4224; CHECK: stocgl %r3, 0(%r2)25; CHECK: br %r1426 %cond = icmp ult i32 %limit, 4227 %orig = load i64, ptr %ptr28 %res = select i1 %cond, i64 %alt, i64 %orig29 store i64 %res, ptr %ptr30 ret void31}32 33; Check the high end of the aligned STOCG range.34define void @f3(ptr %base, i64 %alt, i32 %limit) {35; CHECK-LABEL: f3:36; CHECK: clfi %r4, 4237; CHECK: stocghe %r3, 524280(%r2)38; CHECK: br %r1439 %ptr = getelementptr i64, ptr %base, i64 6553540 %cond = icmp ult i32 %limit, 4241 %orig = load i64, ptr %ptr42 %res = select i1 %cond, i64 %orig, i64 %alt43 store i64 %res, ptr %ptr44 ret void45}46 47; Check the next doubleword up. Other sequences besides this one would be OK.48define void @f4(ptr %base, i64 %alt, i32 %limit) {49; CHECK-LABEL: f4:50; CHECK: agfi %r2, 52428851; CHECK: clfi %r4, 4252; CHECK: stocghe %r3, 0(%r2)53; CHECK: br %r1454 %ptr = getelementptr i64, ptr %base, i64 6553655 %cond = icmp ult i32 %limit, 4256 %orig = load i64, ptr %ptr57 %res = select i1 %cond, i64 %orig, i64 %alt58 store i64 %res, ptr %ptr59 ret void60}61 62; Check the low end of the STOCG range.63define void @f5(ptr %base, i64 %alt, i32 %limit) {64; CHECK-LABEL: f5:65; CHECK: clfi %r4, 4266; CHECK: stocghe %r3, -524288(%r2)67; CHECK: br %r1468 %ptr = getelementptr i64, ptr %base, i64 -6553669 %cond = icmp ult i32 %limit, 4270 %orig = load i64, ptr %ptr71 %res = select i1 %cond, i64 %orig, i64 %alt72 store i64 %res, ptr %ptr73 ret void74}75 76; Check the next doubleword down, with the same comments as f4.77define void @f6(ptr %base, i64 %alt, i32 %limit) {78; CHECK-LABEL: f6:79; CHECK: agfi %r2, -52429680; CHECK: clfi %r4, 4281; CHECK: stocghe %r3, 0(%r2)82; CHECK: br %r1483 %ptr = getelementptr i64, ptr %base, i64 -6553784 %cond = icmp ult i32 %limit, 4285 %orig = load i64, ptr %ptr86 %res = select i1 %cond, i64 %orig, i64 %alt87 store i64 %res, ptr %ptr88 ret void89}90 91; Try a frame index base.92define void @f7(i64 %alt, i32 %limit) {93; CHECK-LABEL: f7:94; CHECK: brasl %r14, foo@PLT95; CHECK: stocghe {{%r[0-9]+}}, {{[0-9]+}}(%r15)96; CHECK: brasl %r14, foo@PLT97; CHECK: br %r1498 %ptr = alloca i6499 call void @foo(ptr %ptr)100 %cond = icmp ult i32 %limit, 42101 %orig = load i64, ptr %ptr102 %res = select i1 %cond, i64 %orig, i64 %alt103 store i64 %res, ptr %ptr104 call void @foo(ptr %ptr)105 ret void106}107 108; Test that conditionally-executed stores do not use STOC, since STOC109; is allowed to trap even when the condition is false.110define void @f8(i64 %a, i64 %b, ptr %dest) {111; CHECK-LABEL: f8:112; CHECK-NOT: stocg %r3, 0(%r4)113; CHECK: br %r14114entry:115 %cmp = icmp ule i64 %a, %b116 br i1 %cmp, label %store, label %exit117 118store:119 store i64 %b, ptr %dest120 br label %exit121 122exit:123 ret void124}125