170 lines · plain
1; Test 64-bit square root.2;3; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z10 \4; RUN: | FileCheck -check-prefix=CHECK -check-prefix=CHECK-SCALAR %s5; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z13 | FileCheck %s6 7declare double @llvm.sqrt.f64(double %f)8declare double @sqrt(double)9 10; Check register square root.11define double @f1(double %val) {12; CHECK-LABEL: f1:13; CHECK: sqdbr %f0, %f014; CHECK: br %r1415 %res = call double @llvm.sqrt.f64(double %val)16 ret double %res17}18 19; Check the low end of the SQDB range.20define double @f2(ptr %ptr) {21; CHECK-LABEL: f2:22; CHECK: sqdb %f0, 0(%r2)23; CHECK: br %r1424 %val = load double, ptr %ptr25 %res = call double @llvm.sqrt.f64(double %val)26 ret double %res27}28 29; Check the high end of the aligned SQDB range.30define double @f3(ptr %base) {31; CHECK-LABEL: f3:32; CHECK: sqdb %f0, 4088(%r2)33; CHECK: br %r1434 %ptr = getelementptr double, ptr %base, i64 51135 %val = load double, ptr %ptr36 %res = call double @llvm.sqrt.f64(double %val)37 ret double %res38}39 40; Check the next doubleword up, which needs separate address logic.41; Other sequences besides this one would be OK.42define double @f4(ptr %base) {43; CHECK-LABEL: f4:44; CHECK: aghi %r2, 409645; CHECK: sqdb %f0, 0(%r2)46; CHECK: br %r1447 %ptr = getelementptr double, ptr %base, i64 51248 %val = load double, ptr %ptr49 %res = call double @llvm.sqrt.f64(double %val)50 ret double %res51}52 53; Check negative displacements, which also need separate address logic.54define double @f5(ptr %base) {55; CHECK-LABEL: f5:56; CHECK: aghi %r2, -857; CHECK: sqdb %f0, 0(%r2)58; CHECK: br %r1459 %ptr = getelementptr double, ptr %base, i64 -160 %val = load double, ptr %ptr61 %res = call double @llvm.sqrt.f64(double %val)62 ret double %res63}64 65; Check that SQDB allows indices.66define double @f6(ptr %base, i64 %index) {67; CHECK-LABEL: f6:68; CHECK: sllg %r1, %r3, 369; CHECK: sqdb %f0, 800(%r1,%r2)70; CHECK: br %r1471 %ptr1 = getelementptr double, ptr %base, i64 %index72 %ptr2 = getelementptr double, ptr %ptr1, i64 10073 %val = load double, ptr %ptr274 %res = call double @llvm.sqrt.f64(double %val)75 ret double %res76}77 78; Test a case where we spill the source of at least one SQDBR. We want79; to use SQDB if possible.80define void @f7(ptr %ptr) {81; CHECK-LABEL: f7:82; CHECK-SCALAR: sqdb {{%f[0-9]+}}, 160(%r15)83; CHECK: br %r1484 %val0 = load volatile double, ptr %ptr85 %val1 = load volatile double, ptr %ptr86 %val2 = load volatile double, ptr %ptr87 %val3 = load volatile double, ptr %ptr88 %val4 = load volatile double, ptr %ptr89 %val5 = load volatile double, ptr %ptr90 %val6 = load volatile double, ptr %ptr91 %val7 = load volatile double, ptr %ptr92 %val8 = load volatile double, ptr %ptr93 %val9 = load volatile double, ptr %ptr94 %val10 = load volatile double, ptr %ptr95 %val11 = load volatile double, ptr %ptr96 %val12 = load volatile double, ptr %ptr97 %val13 = load volatile double, ptr %ptr98 %val14 = load volatile double, ptr %ptr99 %val15 = load volatile double, ptr %ptr100 %val16 = load volatile double, ptr %ptr101 102 %sqrt0 = call double @llvm.sqrt.f64(double %val0)103 %sqrt1 = call double @llvm.sqrt.f64(double %val1)104 %sqrt2 = call double @llvm.sqrt.f64(double %val2)105 %sqrt3 = call double @llvm.sqrt.f64(double %val3)106 %sqrt4 = call double @llvm.sqrt.f64(double %val4)107 %sqrt5 = call double @llvm.sqrt.f64(double %val5)108 %sqrt6 = call double @llvm.sqrt.f64(double %val6)109 %sqrt7 = call double @llvm.sqrt.f64(double %val7)110 %sqrt8 = call double @llvm.sqrt.f64(double %val8)111 %sqrt9 = call double @llvm.sqrt.f64(double %val9)112 %sqrt10 = call double @llvm.sqrt.f64(double %val10)113 %sqrt11 = call double @llvm.sqrt.f64(double %val11)114 %sqrt12 = call double @llvm.sqrt.f64(double %val12)115 %sqrt13 = call double @llvm.sqrt.f64(double %val13)116 %sqrt14 = call double @llvm.sqrt.f64(double %val14)117 %sqrt15 = call double @llvm.sqrt.f64(double %val15)118 %sqrt16 = call double @llvm.sqrt.f64(double %val16)119 120 store volatile double %val0, ptr %ptr121 store volatile double %val1, ptr %ptr122 store volatile double %val2, ptr %ptr123 store volatile double %val3, ptr %ptr124 store volatile double %val4, ptr %ptr125 store volatile double %val5, ptr %ptr126 store volatile double %val6, ptr %ptr127 store volatile double %val7, ptr %ptr128 store volatile double %val8, ptr %ptr129 store volatile double %val9, ptr %ptr130 store volatile double %val10, ptr %ptr131 store volatile double %val11, ptr %ptr132 store volatile double %val12, ptr %ptr133 store volatile double %val13, ptr %ptr134 store volatile double %val14, ptr %ptr135 store volatile double %val15, ptr %ptr136 store volatile double %val16, ptr %ptr137 138 store volatile double %sqrt0, ptr %ptr139 store volatile double %sqrt1, ptr %ptr140 store volatile double %sqrt2, ptr %ptr141 store volatile double %sqrt3, ptr %ptr142 store volatile double %sqrt4, ptr %ptr143 store volatile double %sqrt5, ptr %ptr144 store volatile double %sqrt6, ptr %ptr145 store volatile double %sqrt7, ptr %ptr146 store volatile double %sqrt8, ptr %ptr147 store volatile double %sqrt9, ptr %ptr148 store volatile double %sqrt10, ptr %ptr149 store volatile double %sqrt11, ptr %ptr150 store volatile double %sqrt12, ptr %ptr151 store volatile double %sqrt13, ptr %ptr152 store volatile double %sqrt14, ptr %ptr153 store volatile double %sqrt15, ptr %ptr154 store volatile double %sqrt16, ptr %ptr155 156 ret void157}158 159; Check that a call to the normal sqrt function is lowered.160define double @f8(double %dummy, double %val) {161; CHECK-LABEL: f8:162; CHECK: sqdbr %f0, %f2163; CHECK: cdbr %f0, %f0164; CHECK: bnor %r14165; CHECK: ldr %f0, %f2166; CHECK: jg sqrt@PLT167 %res = tail call double @sqrt(double %val)168 ret double %res169}170