Index: clang/include/clang/Basic/CodeGenOptions.h =================================================================== --- clang/include/clang/Basic/CodeGenOptions.h +++ clang/include/clang/Basic/CodeGenOptions.h @@ -55,13 +55,14 @@ }; enum VectorLibrary { - NoLibrary, // Don't use any vector library. - Accelerate, // Use the Accelerate framework. - LIBMVEC, // GLIBC vector math library. - MASSV, // IBM MASS vector library. - SVML, // Intel short vector math library. - SLEEF, // SLEEF SIMD Library for Evaluating Elementary Functions. - Darwin_libsystem_m // Use Darwin's libsytem_m vector functions. + NoLibrary, // Don't use any vector library. + Accelerate, // Use the Accelerate framework. + LIBMVEC, // GLIBC vector math library. + MASSV, // IBM MASS vector library. + SVML, // Intel short vector math library. + SLEEF, // SLEEF SIMD Library for Evaluating Elementary Functions. + Darwin_libsystem_m, // Use Darwin's libsytem_m vector functions. + ArmPL // Arm Performance Libraries. }; enum ObjCDispatchMethodKind { Index: clang/include/clang/Driver/Options.td =================================================================== --- clang/include/clang/Driver/Options.td +++ clang/include/clang/Driver/Options.td @@ -2612,10 +2612,10 @@ Alias; def fveclib : Joined<["-"], "fveclib=">, Group, Flags<[CC1Option]>, HelpText<"Use the given vector functions library">, - Values<"Accelerate,libmvec,MASSV,SVML,SLEEF,Darwin_libsystem_m,none">, + Values<"Accelerate,libmvec,MASSV,SVML,SLEEF,Darwin_libsystem_m,ArmPL,none">, NormalizedValuesScope<"CodeGenOptions">, NormalizedValues<["Accelerate", "LIBMVEC", "MASSV", "SVML", "SLEEF", - "Darwin_libsystem_m", "NoLibrary"]>, + "Darwin_libsystem_m", "ArmPL", "NoLibrary"]>, MarshallingInfoEnum, "NoLibrary">; def fno_lax_vector_conversions : Flag<["-"], "fno-lax-vector-conversions">, Group, Alias, AliasArgs<["none"]>; Index: clang/lib/CodeGen/BackendUtil.cpp =================================================================== --- clang/lib/CodeGen/BackendUtil.cpp +++ clang/lib/CodeGen/BackendUtil.cpp @@ -284,6 +284,10 @@ TLII->addVectorizableFunctionsFromVecLib( TargetLibraryInfoImpl::DarwinLibSystemM, TargetTriple); break; + case CodeGenOptions::ArmPL: + TLII->addVectorizableFunctionsFromVecLib(TargetLibraryInfoImpl::ArmPL, + TargetTriple); + break; default: break; } Index: clang/lib/Driver/ToolChains/Clang.cpp =================================================================== --- clang/lib/Driver/ToolChains/Clang.cpp +++ clang/lib/Driver/ToolChains/Clang.cpp @@ -5300,7 +5300,7 @@ Triple.getArch() != llvm::Triple::x86_64) D.Diag(diag::err_drv_unsupported_opt_for_target) << Name << Triple.getArchName(); - } else if (Name == "SLEEF") { + } else if (Name == "SLEEF" || Name == "ArmPL") { if (Triple.getArch() != llvm::Triple::aarch64 && Triple.getArch() != llvm::Triple::aarch64_be) D.Diag(diag::err_drv_unsupported_opt_for_target) Index: clang/test/Driver/autocomplete.c =================================================================== --- clang/test/Driver/autocomplete.c +++ clang/test/Driver/autocomplete.c @@ -80,6 +80,7 @@ // FLTOALL-NEXT: thin // RUN: %clang --autocomplete=-fveclib= | FileCheck %s -check-prefix=FVECLIBALL // FVECLIBALL: Accelerate +// FVECLIBALL-NEXT: ArmPL // FVECLIBALL-NEXT: Darwin_libsystem_m // FVECLIBALL-NEXT: libmvec // FVECLIBALL-NEXT: MASSV Index: clang/test/Driver/fveclib.c =================================================================== --- clang/test/Driver/fveclib.c +++ clang/test/Driver/fveclib.c @@ -4,6 +4,7 @@ // RUN: %clang -### -c -fveclib=MASSV %s 2>&1 | FileCheck -check-prefix CHECK-MASSV %s // RUN: %clang -### -c -fveclib=Darwin_libsystem_m %s 2>&1 | FileCheck -check-prefix CHECK-DARWIN_LIBSYSTEM_M %s // RUN: %clang -### -c --target=aarch64-none-none -fveclib=SLEEF %s 2>&1 | FileCheck -check-prefix CHECK-SLEEF %s +// RUN: %clang -### -c --target=aarch64-none-none -fveclib=ArmPL %s 2>&1 | FileCheck -check-prefix CHECK-ARMPL %s // RUN: not %clang -c -fveclib=something %s 2>&1 | FileCheck -check-prefix CHECK-INVALID %s // CHECK-NOLIB: "-fveclib=none" @@ -12,10 +13,12 @@ // CHECK-MASSV: "-fveclib=MASSV" // CHECK-DARWIN_LIBSYSTEM_M: "-fveclib=Darwin_libsystem_m" // CHECK-SLEEF: "-fveclib=SLEEF" +// CHECK-ARMPL: "-fveclib=ArmPL" // CHECK-INVALID: error: invalid value 'something' in '-fveclib=something' // RUN: not %clang --target=x86-none-none -c -fveclib=SLEEF %s 2>&1 | FileCheck -check-prefix CHECK-ERROR %s +// RUN: not %clang --target=x86-none-none -c -fveclib=ArmPL %s 2>&1 | FileCheck -check-prefix CHECK-ERROR %s // RUN: not %clang --target=aarch64-none-none -c -fveclib=LIBMVEC-X86 %s 2>&1 | FileCheck -check-prefix CHECK-ERROR %s // RUN: not %clang --target=aarch64-none-none -c -fveclib=SVML %s 2>&1 | FileCheck -check-prefix CHECK-ERROR %s // CHECK-ERROR: unsupported option {{.*}} for target Index: llvm/include/llvm/Analysis/TargetLibraryInfo.h =================================================================== --- llvm/include/llvm/Analysis/TargetLibraryInfo.h +++ llvm/include/llvm/Analysis/TargetLibraryInfo.h @@ -96,7 +96,8 @@ LIBMVEC_X86, // GLIBC Vector Math library. MASSV, // IBM MASS vector library. SVML, // Intel short vector math library. - SLEEFGNUABI // SLEEF - SIMD Library for Evaluating Elementary Functions. + SLEEFGNUABI, // SLEEF - SIMD Library for Evaluating Elementary Functions. + ArmPL // Arm Performance Libraries. }; TargetLibraryInfoImpl(); Index: llvm/include/llvm/Analysis/VecFuncs.def =================================================================== --- llvm/include/llvm/Analysis/VecFuncs.def +++ llvm/include/llvm/Analysis/VecFuncs.def @@ -682,6 +682,253 @@ TLI_DEFINE_VECFUNC("tgamma", "_ZGVsMxv_tgamma", SCALABLE(2), MASKED) TLI_DEFINE_VECFUNC("tgammaf", "_ZGVsMxv_tgammaf", SCALABLE(4), MASKED) +#elif defined(TLI_DEFINE_ARMPL_VECFUNCS) + +TLI_DEFINE_VECFUNC("acos", "armpl_vacosq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("acosf", "armpl_vacosq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("acos", "armpl_svacos_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("acosf", "armpl_svacos_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("acosh", "armpl_vacoshq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("acoshf", "armpl_vacoshq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("acosh", "armpl_svacosh_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("acoshf", "armpl_svacosh_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("asin", "armpl_vasinq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("asinf", "armpl_vasinq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("asin", "armpl_svasin_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("asinf", "armpl_svasin_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("asinh", "armpl_vasinhq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("asinhf", "armpl_vasinhq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("asinh", "armpl_svasinh_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("asinhf", "armpl_svasinh_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("atan", "armpl_vatanq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("atanf", "armpl_vatanq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("atan", "armpl_svatan_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("atanf", "armpl_svatan_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("atan2", "armpl_vatan2q_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("atan2f", "armpl_vatan2q_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("atan2", "armpl_svatan2_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("atan2f", "armpl_svatan2_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("atanh", "armpl_vatanhq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("atanhf", "armpl_vatanhq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("atanh", "armpl_svatanh_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("atanhf", "armpl_svatanh_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("cbrt", "armpl_vcbrtq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("cbrtf", "armpl_vcbrtq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("cbrt", "armpl_svcbrt_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("cbrtf", "armpl_svcbrt_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("copysign", "armpl_vcopysignq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("copysignf", "armpl_vcopysignq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("copysign", "armpl_svcopysign_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("copysignf", "armpl_svcopysign_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("cos", "armpl_vcosq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("cosf", "armpl_vcosq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("cos", "armpl_svcos_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("cosf", "armpl_svcos_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("llvm.cos.f64", "armpl_vcosq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("llvm.cos.f32", "armpl_vcosq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("llvm.cos.f64", "armpl_svcos_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("llvm.cos.f32", "armpl_svcos_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("cosh", "armpl_vcoshq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("coshf", "armpl_vcoshq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("cosh", "armpl_svcosh_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("coshf", "armpl_svcosh_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("erf", "armpl_verfq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("erff", "armpl_verfq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("erf", "armpl_sverf_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("erff", "armpl_sverf_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("erfc", "armpl_verfcq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("erfcf", "armpl_verfcq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("erfc", "armpl_sverfc_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("erfcf", "armpl_sverfc_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("exp", "armpl_vexpq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("expf", "armpl_vexpq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("exp", "armpl_svexp_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("expf", "armpl_svexp_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("llvm.exp.f64", "armpl_vexpq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("llvm.exp.f32", "armpl_vexpq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("llvm.exp.f64", "armpl_svexp_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("llvm.exp.f32", "armpl_svexp_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("exp2", "armpl_vexp2q_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("exp2f", "armpl_vexp2q_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("exp2", "armpl_svexp2_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("exp2f", "armpl_svexp2_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("llvm.exp2.f64", "armpl_vexp2q_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("llvm.exp2.f32", "armpl_vexp2q_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("llvm.exp2.f64", "armpl_svexp2_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("llvm.exp2.f32", "armpl_svexp2_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("exp10", "armpl_vexp10q_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("exp10f", "armpl_vexp10q_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("exp10", "armpl_svexp10_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("exp10f", "armpl_svexp10_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("expm1", "armpl_vexpm1q_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("expm1f", "armpl_vexpm1q_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("expm1", "armpl_svexpm1_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("expm1f", "armpl_svexpm1_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("fdim", "armpl_vfdimq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("fdimf", "armpl_vfdimq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("fdim", "armpl_svfdim_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("fdimf", "armpl_svfdim_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("fma", "armpl_vfmaq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("fmaf", "armpl_vfmaq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("fma", "armpl_svfma_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("fmaf", "armpl_svfma_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("fmin", "armpl_vfminq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("fminf", "armpl_vfminq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("fmin", "armpl_svfmin_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("fminf", "armpl_svfmin_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("fmod", "armpl_vfmodq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("fmodf", "armpl_vfmodq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("fmod", "armpl_svfmod_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("fmodf", "armpl_svfmod_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("hypot", "armpl_vhypotq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("hypotf", "armpl_vhypotq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("hypot", "armpl_svhypot_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("hypotf", "armpl_svhypot_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("ilogb", "armpl_vilogbq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("ilogbf", "armpl_vilogbq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("ilogb", "armpl_svilogb_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("ilogbf", "armpl_svilogb_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("ldexp", "armpl_vldexpq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("ldexpf", "armpl_vldexpq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("ldexp", "armpl_svldexp_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("ldexpf", "armpl_svldexp_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("lgamma", "armpl_vlgammaq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("lgammaf", "armpl_vlgammaq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("lgamma", "armpl_svlgamma_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("lgammaf", "armpl_svlgamma_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("log", "armpl_vlogq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("logf", "armpl_vlogq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("log", "armpl_svlog_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("logf", "armpl_svlog_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("llvm.log.f64", "armpl_vlogq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("llvm.log.f32", "armpl_vlogq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("llvm.log.f64", "armpl_svlog_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("llvm.log.f32", "armpl_svlog_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("log1p", "armpl_vlog1pq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("log1pf", "armpl_vlog1pq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("log1p", "armpl_svlog1p_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("log1pf", "armpl_svlog1p_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("log2", "armpl_vlog2q_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("log2f", "armpl_vlog2q_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("log2", "armpl_svlog2_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("log2f", "armpl_svlog2_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("llvm.log2.f64", "armpl_vlog2q_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("llvm.log2.f32", "armpl_vlog2q_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("llvm.log2.f64", "armpl_svlog2_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("llvm.log2.f32", "armpl_svlog2_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("log10", "armpl_vlog10q_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("log10f", "armpl_vlog10q_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("log10", "armpl_svlog10_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("log10f", "armpl_svlog10_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("llvm.log10.f64", "armpl_vlog10q_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("llvm.log10.f32", "armpl_vlog10q_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("llvm.log10.f64", "armpl_svlog10_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("llvm.log10.f32", "armpl_svlog10_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("modf", "armpl_vmodfq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("modff", "armpl_vmodfq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("modf", "armpl_svmodf_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("modff", "armpl_svmodf_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("nextafter", "armpl_vnextafterq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("nextafterf", "armpl_vnextafterq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("nextafter", "armpl_svnextafter_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("nextafterf", "armpl_svnextafter_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("pow", "armpl_vpowq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("powf", "armpl_vpowq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("pow", "armpl_svpow_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("powf", "armpl_svpow_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("llvm.pow.f64", "armpl_vpowq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("llvm.pow.f32", "armpl_vpowq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("llvm.pow.f64", "armpl_svpow_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("llvm.pow.f32", "armpl_svpow_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("sin", "armpl_vsinq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("sinf", "armpl_vsinq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("sin", "armpl_svsin_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("sinf", "armpl_svsin_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("llvm.sin.f64", "armpl_vsinq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("llvm.sin.f32", "armpl_vsinq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("llvm.sin.f64", "armpl_svsin_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("llvm.sin.f32", "armpl_svsin_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("sincos", "armpl_vsincosq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("sincosf", "armpl_vsincosq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("sincos", "armpl_svsincos_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("sincosf", "armpl_svsincos_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("sincospi", "armpl_vsincospiq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("sincospif", "armpl_vsincospiq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("sincospi", "armpl_svsincospi_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("sincospif", "armpl_svsincospi_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("sinh", "armpl_vsinhq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("sinhf", "armpl_vsinhq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("sinh", "armpl_svsinh_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("sinhf", "armpl_svsinh_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("sinpi", "armpl_vsinpiq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("sinpif", "armpl_vsinpiq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("sinpi", "armpl_svsinpi_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("sinpif", "armpl_svsinpi_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("sqrt", "armpl_vsqrtq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("sqrtf", "armpl_vsqrtq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("sqrt", "armpl_svsqrt_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("sqrtf", "armpl_svsqrt_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("tan", "armpl_vtanq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("tanf", "armpl_vtanq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("tan", "armpl_svtan_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("tanf", "armpl_svtan_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("tanh", "armpl_vtanhq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("tanhf", "armpl_vtanhq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("tanh", "armpl_svtanh_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("tanhf", "armpl_svtanh_f32_x", SCALABLE(4), MASKED) + +TLI_DEFINE_VECFUNC("tgamma", "armpl_vtgammaq_f64", FIXED(2), NOMASK) +TLI_DEFINE_VECFUNC("tgammaf", "armpl_vtgammaq_f32", FIXED(4), NOMASK) +TLI_DEFINE_VECFUNC("tgamma", "armpl_svtgamma_f64_x", SCALABLE(2), MASKED) +TLI_DEFINE_VECFUNC("tgammaf", "armpl_svtgamma_f32_x", SCALABLE(4), MASKED) + #else #error "Must choose which vector library functions are to be defined." #endif @@ -701,3 +948,4 @@ #undef TLI_DEFINE_SLEEFGNUABI_VF4_VECFUNCS #undef TLI_DEFINE_SLEEFGNUABI_SCALABLE_VECFUNCS #undef TLI_DEFINE_MASSV_VECFUNCS_NAMES +#undef TLI_DEFINE_ARMPL_VECFUNCS Index: llvm/lib/Analysis/TargetLibraryInfo.cpp =================================================================== --- llvm/lib/Analysis/TargetLibraryInfo.cpp +++ llvm/lib/Analysis/TargetLibraryInfo.cpp @@ -33,7 +33,9 @@ clEnumValN(TargetLibraryInfoImpl::SVML, "SVML", "Intel SVML library"), clEnumValN(TargetLibraryInfoImpl::SLEEFGNUABI, "sleefgnuabi", - "SIMD Library for Evaluating Elementary Functions"))); + "SIMD Library for Evaluating Elementary Functions"), + clEnumValN(TargetLibraryInfoImpl::ArmPL, "ArmPL", + "Arm Performance Libraries"))); StringLiteral const TargetLibraryInfoImpl::StandardNames[LibFunc::NumLibFuncs] = { @@ -1215,6 +1217,23 @@ } break; } + case ArmPL: { + const VecDesc VecFuncs[] = { +#define TLI_DEFINE_ARMPL_VECFUNCS +#define TLI_DEFINE_VECFUNC(SCAL, VEC, VF, MASK) {SCAL, VEC, VF, MASK}, +#include "llvm/Analysis/VecFuncs.def" + }; + + switch (TargetTriple.getArch()) { + default: + break; + case llvm::Triple::aarch64: + case llvm::Triple::aarch64_be: + addVectorizableFunctions(VecFuncs); + break; + } + break; + } case NoLibrary: break; } Index: llvm/test/CodeGen/AArch64/replace-intrinsics-with-veclib-armpl.ll =================================================================== --- /dev/null +++ llvm/test/CodeGen/AArch64/replace-intrinsics-with-veclib-armpl.ll @@ -0,0 +1,539 @@ +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2 +; RUN: opt -S -vector-library=ArmPL -replace-with-veclib < %s | FileCheck %s --check-prefixes=NEON +; RUN: opt -S -mattr=+sve -vector-library=ArmPL -replace-with-veclib < %s | FileCheck %s --check-prefixes=SVE + +target triple = "aarch64-unknown-linux-gnu" + +; +; The replace-with-veclib pass does not work with scalable types, thus +; the mappings aren't utilised. Tests will need to be regenerated when the +; pass is improved. +; + +declare <2 x double> @llvm.cos.v2f64(<2 x double>) +declare <4 x float> @llvm.cos.v4f32(<4 x float>) +declare @llvm.cos.nxv2f64() +declare @llvm.cos.nxv4f32() + +define <2 x double> @llvm_cos_f64(<2 x double> %in) { +; NEON-LABEL: define <2 x double> @llvm_cos_f64 +; NEON-SAME: (<2 x double> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vcosq_f64(<2 x double> [[IN]]) +; NEON-NEXT: ret <2 x double> [[TMP1]] +; +; SVE-LABEL: define <2 x double> @llvm_cos_f64 +; SVE-SAME: (<2 x double> [[IN:%.*]]) #[[ATTR1:[0-9]+]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vcosq_f64(<2 x double> [[IN]]) +; SVE-NEXT: ret <2 x double> [[TMP1]] +; + %1 = call fast <2 x double> @llvm.cos.v2f64(<2 x double> %in) + ret <2 x double> %1 +} + +define <4 x float> @llvm_cos_f32(<4 x float> %in) { +; NEON-LABEL: define <4 x float> @llvm_cos_f32 +; NEON-SAME: (<4 x float> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vcosq_f32(<4 x float> [[IN]]) +; NEON-NEXT: ret <4 x float> [[TMP1]] +; +; SVE-LABEL: define <4 x float> @llvm_cos_f32 +; SVE-SAME: (<4 x float> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vcosq_f32(<4 x float> [[IN]]) +; SVE-NEXT: ret <4 x float> [[TMP1]] +; + %1 = call fast <4 x float> @llvm.cos.v4f32(<4 x float> %in) + ret <4 x float> %1 +} + +define @llvm_cos_vscale_f64( %in) { +; NEON-LABEL: define @llvm_cos_vscale_f64 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.cos.nxv2f64( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_cos_vscale_f64 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.cos.nxv2f64( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.cos.nxv2f64( %in) + ret %1 +} + +define @llvm_cos_vscale_f32( %in) { +; NEON-LABEL: define @llvm_cos_vscale_f32 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.cos.nxv4f32( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_cos_vscale_f32 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.cos.nxv4f32( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.cos.nxv4f32( %in) + ret %1 +} + +declare <2 x double> @llvm.sin.v2f64(<2 x double>) +declare <4 x float> @llvm.sin.v4f32(<4 x float>) +declare @llvm.sin.nxv2f64() +declare @llvm.sin.nxv4f32() + +define <2 x double> @llvm_sin_f64(<2 x double> %in) { +; NEON-LABEL: define <2 x double> @llvm_sin_f64 +; NEON-SAME: (<2 x double> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vsinq_f64(<2 x double> [[IN]]) +; NEON-NEXT: ret <2 x double> [[TMP1]] +; +; SVE-LABEL: define <2 x double> @llvm_sin_f64 +; SVE-SAME: (<2 x double> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vsinq_f64(<2 x double> [[IN]]) +; SVE-NEXT: ret <2 x double> [[TMP1]] +; + %1 = call fast <2 x double> @llvm.sin.v2f64(<2 x double> %in) + ret <2 x double> %1 +} + +define <4 x float> @llvm_sin_f32(<4 x float> %in) { +; NEON-LABEL: define <4 x float> @llvm_sin_f32 +; NEON-SAME: (<4 x float> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vsinq_f32(<4 x float> [[IN]]) +; NEON-NEXT: ret <4 x float> [[TMP1]] +; +; SVE-LABEL: define <4 x float> @llvm_sin_f32 +; SVE-SAME: (<4 x float> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vsinq_f32(<4 x float> [[IN]]) +; SVE-NEXT: ret <4 x float> [[TMP1]] +; + %1 = call fast <4 x float> @llvm.sin.v4f32(<4 x float> %in) + ret <4 x float> %1 +} + +define @llvm_sin_vscale_f64( %in) { +; NEON-LABEL: define @llvm_sin_vscale_f64 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.sin.nxv2f64( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_sin_vscale_f64 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.sin.nxv2f64( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.sin.nxv2f64( %in) + ret %1 +} + +define @llvm_sin_vscale_f32( %in) { +; NEON-LABEL: define @llvm_sin_vscale_f32 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.sin.nxv4f32( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_sin_vscale_f32 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.sin.nxv4f32( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.sin.nxv4f32( %in) + ret %1 +} + +declare <2 x double> @llvm.exp.v2f64(<2 x double>) +declare <4 x float> @llvm.exp.v4f32(<4 x float>) +declare @llvm.exp.nxv2f64() +declare @llvm.exp.nxv4f32() + +define <2 x double> @llvm_exp_f64(<2 x double> %in) { +; NEON-LABEL: define <2 x double> @llvm_exp_f64 +; NEON-SAME: (<2 x double> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vexpq_f64(<2 x double> [[IN]]) +; NEON-NEXT: ret <2 x double> [[TMP1]] +; +; SVE-LABEL: define <2 x double> @llvm_exp_f64 +; SVE-SAME: (<2 x double> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vexpq_f64(<2 x double> [[IN]]) +; SVE-NEXT: ret <2 x double> [[TMP1]] +; + %1 = call fast <2 x double> @llvm.exp.v2f64(<2 x double> %in) + ret <2 x double> %1 +} + +define <4 x float> @llvm_exp_f32(<4 x float> %in) { +; NEON-LABEL: define <4 x float> @llvm_exp_f32 +; NEON-SAME: (<4 x float> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vexpq_f32(<4 x float> [[IN]]) +; NEON-NEXT: ret <4 x float> [[TMP1]] +; +; SVE-LABEL: define <4 x float> @llvm_exp_f32 +; SVE-SAME: (<4 x float> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vexpq_f32(<4 x float> [[IN]]) +; SVE-NEXT: ret <4 x float> [[TMP1]] +; + %1 = call fast <4 x float> @llvm.exp.v4f32(<4 x float> %in) + ret <4 x float> %1 +} + +define @llvm_exp_vscale_f64( %in) { +; NEON-LABEL: define @llvm_exp_vscale_f64 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.exp.nxv2f64( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_exp_vscale_f64 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.exp.nxv2f64( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.exp.nxv2f64( %in) + ret %1 +} + +define @llvm_exp_vscale_f32( %in) { +; NEON-LABEL: define @llvm_exp_vscale_f32 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.exp.nxv4f32( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_exp_vscale_f32 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.exp.nxv4f32( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.exp.nxv4f32( %in) + ret %1 +} + +declare <2 x double> @llvm.exp2.v2f64(<2 x double>) +declare <4 x float> @llvm.exp2.v4f32(<4 x float>) +declare @llvm.exp2.nxv2f64() +declare @llvm.exp2.nxv4f32() + +define <2 x double> @llvm_exp2_f64(<2 x double> %in) { +; NEON-LABEL: define <2 x double> @llvm_exp2_f64 +; NEON-SAME: (<2 x double> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vexp2q_f64(<2 x double> [[IN]]) +; NEON-NEXT: ret <2 x double> [[TMP1]] +; +; SVE-LABEL: define <2 x double> @llvm_exp2_f64 +; SVE-SAME: (<2 x double> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vexp2q_f64(<2 x double> [[IN]]) +; SVE-NEXT: ret <2 x double> [[TMP1]] +; + %1 = call fast <2 x double> @llvm.exp2.v2f64(<2 x double> %in) + ret <2 x double> %1 +} + +define <4 x float> @llvm_exp2_f32(<4 x float> %in) { +; NEON-LABEL: define <4 x float> @llvm_exp2_f32 +; NEON-SAME: (<4 x float> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vexp2q_f32(<4 x float> [[IN]]) +; NEON-NEXT: ret <4 x float> [[TMP1]] +; +; SVE-LABEL: define <4 x float> @llvm_exp2_f32 +; SVE-SAME: (<4 x float> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vexp2q_f32(<4 x float> [[IN]]) +; SVE-NEXT: ret <4 x float> [[TMP1]] +; + %1 = call fast <4 x float> @llvm.exp2.v4f32(<4 x float> %in) + ret <4 x float> %1 +} + +define @llvm_exp2_vscale_f64( %in) { +; NEON-LABEL: define @llvm_exp2_vscale_f64 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.exp2.nxv2f64( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_exp2_vscale_f64 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.exp2.nxv2f64( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.exp2.nxv2f64( %in) + ret %1 +} + +define @llvm_exp2_vscale_f32( %in) { +; NEON-LABEL: define @llvm_exp2_vscale_f32 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.exp2.nxv4f32( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_exp2_vscale_f32 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.exp2.nxv4f32( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.exp2.nxv4f32( %in) + ret %1 +} + + +declare <2 x double> @llvm.log.v2f64(<2 x double>) +declare <4 x float> @llvm.log.v4f32(<4 x float>) +declare @llvm.log.nxv2f64() +declare @llvm.log.nxv4f32() + +define <2 x double> @llvm_log_f64(<2 x double> %in) { +; NEON-LABEL: define <2 x double> @llvm_log_f64 +; NEON-SAME: (<2 x double> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vlogq_f64(<2 x double> [[IN]]) +; NEON-NEXT: ret <2 x double> [[TMP1]] +; +; SVE-LABEL: define <2 x double> @llvm_log_f64 +; SVE-SAME: (<2 x double> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vlogq_f64(<2 x double> [[IN]]) +; SVE-NEXT: ret <2 x double> [[TMP1]] +; + %1 = call fast <2 x double> @llvm.log.v2f64(<2 x double> %in) + ret <2 x double> %1 +} + +define <4 x float> @llvm_log_f32(<4 x float> %in) { +; NEON-LABEL: define <4 x float> @llvm_log_f32 +; NEON-SAME: (<4 x float> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vlogq_f32(<4 x float> [[IN]]) +; NEON-NEXT: ret <4 x float> [[TMP1]] +; +; SVE-LABEL: define <4 x float> @llvm_log_f32 +; SVE-SAME: (<4 x float> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vlogq_f32(<4 x float> [[IN]]) +; SVE-NEXT: ret <4 x float> [[TMP1]] +; + %1 = call fast <4 x float> @llvm.log.v4f32(<4 x float> %in) + ret <4 x float> %1 +} + +define @llvm_log_vscale_f64( %in) { +; NEON-LABEL: define @llvm_log_vscale_f64 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.log.nxv2f64( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_log_vscale_f64 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.log.nxv2f64( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.log.nxv2f64( %in) + ret %1 +} + +define @llvm_log_vscale_f32( %in) { +; NEON-LABEL: define @llvm_log_vscale_f32 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.log.nxv4f32( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_log_vscale_f32 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.log.nxv4f32( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.log.nxv4f32( %in) + ret %1 +} + +declare <2 x double> @llvm.log2.v2f64(<2 x double>) +declare <4 x float> @llvm.log2.v4f32(<4 x float>) +declare @llvm.log2.nxv2f64() +declare @llvm.log2.nxv4f32() + +define <2 x double> @llvm_log2_f64(<2 x double> %in) { +; NEON-LABEL: define <2 x double> @llvm_log2_f64 +; NEON-SAME: (<2 x double> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vlog2q_f64(<2 x double> [[IN]]) +; NEON-NEXT: ret <2 x double> [[TMP1]] +; +; SVE-LABEL: define <2 x double> @llvm_log2_f64 +; SVE-SAME: (<2 x double> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vlog2q_f64(<2 x double> [[IN]]) +; SVE-NEXT: ret <2 x double> [[TMP1]] +; + %1 = call fast <2 x double> @llvm.log2.v2f64(<2 x double> %in) + ret <2 x double> %1 +} + +define <4 x float> @llvm_log2_f32(<4 x float> %in) { +; NEON-LABEL: define <4 x float> @llvm_log2_f32 +; NEON-SAME: (<4 x float> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vlog2q_f32(<4 x float> [[IN]]) +; NEON-NEXT: ret <4 x float> [[TMP1]] +; +; SVE-LABEL: define <4 x float> @llvm_log2_f32 +; SVE-SAME: (<4 x float> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vlog2q_f32(<4 x float> [[IN]]) +; SVE-NEXT: ret <4 x float> [[TMP1]] +; + %1 = call fast <4 x float> @llvm.log2.v4f32(<4 x float> %in) + ret <4 x float> %1 +} + +define @llvm_log2_vscale_f64( %in) { +; NEON-LABEL: define @llvm_log2_vscale_f64 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.log2.nxv2f64( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_log2_vscale_f64 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.log2.nxv2f64( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.log2.nxv2f64( %in) + ret %1 +} + +define @llvm_log2_vscale_f32( %in) { +; NEON-LABEL: define @llvm_log2_vscale_f32 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.log2.nxv4f32( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_log2_vscale_f32 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.log2.nxv4f32( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.log2.nxv4f32( %in) + ret %1 +} + +declare <2 x double> @llvm.log10.v2f64(<2 x double>) +declare <4 x float> @llvm.log10.v4f32(<4 x float>) +declare @llvm.log10.nxv2f64() +declare @llvm.log10.nxv4f32() + +define <2 x double> @llvm_log10_f64(<2 x double> %in) { +; NEON-LABEL: define <2 x double> @llvm_log10_f64 +; NEON-SAME: (<2 x double> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vlog10q_f64(<2 x double> [[IN]]) +; NEON-NEXT: ret <2 x double> [[TMP1]] +; +; SVE-LABEL: define <2 x double> @llvm_log10_f64 +; SVE-SAME: (<2 x double> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <2 x double> @armpl_vlog10q_f64(<2 x double> [[IN]]) +; SVE-NEXT: ret <2 x double> [[TMP1]] +; + %1 = call fast <2 x double> @llvm.log10.v2f64(<2 x double> %in) + ret <2 x double> %1 +} + +define <4 x float> @llvm_log10_f32(<4 x float> %in) { +; NEON-LABEL: define <4 x float> @llvm_log10_f32 +; NEON-SAME: (<4 x float> [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vlog10q_f32(<4 x float> [[IN]]) +; NEON-NEXT: ret <4 x float> [[TMP1]] +; +; SVE-LABEL: define <4 x float> @llvm_log10_f32 +; SVE-SAME: (<4 x float> [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <4 x float> @armpl_vlog10q_f32(<4 x float> [[IN]]) +; SVE-NEXT: ret <4 x float> [[TMP1]] +; + %1 = call fast <4 x float> @llvm.log10.v4f32(<4 x float> %in) + ret <4 x float> %1 +} + +define @llvm_log10_vscale_f64( %in) { +; NEON-LABEL: define @llvm_log10_vscale_f64 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.log10.nxv2f64( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_log10_vscale_f64 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.log10.nxv2f64( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.log10.nxv2f64( %in) + ret %1 +} + +define @llvm_log10_vscale_f32( %in) { +; NEON-LABEL: define @llvm_log10_vscale_f32 +; NEON-SAME: ( [[IN:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.log10.nxv4f32( [[IN]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_log10_vscale_f32 +; SVE-SAME: ( [[IN:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.log10.nxv4f32( [[IN]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.log10.nxv4f32( %in) + ret %1 +} + +declare <2 x double> @llvm.pow.v2f64(<2 x double>, <2 x double>) +declare <4 x float> @llvm.pow.v4f32(<4 x float>, <4 x float>) +declare @llvm.pow.nxv2f64(, ) +declare @llvm.pow.nxv4f32(, ) + +; +; There is a bug in the replace-with-veclib pass, and for intrinsics which take +; more than one arguments, but has just one overloaded type, it incorrectly +; reconstructs the scalar name, for pow specificlly it is searching for: +; llvm.pow.f64.f64 and llvm.pow.f32.f32 +; + +define <2 x double> @llvm_pow_f64(<2 x double> %in, <2 x double> %power) { +; NEON-LABEL: define <2 x double> @llvm_pow_f64 +; NEON-SAME: (<2 x double> [[IN:%.*]], <2 x double> [[POWER:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <2 x double> @llvm.pow.v2f64(<2 x double> [[IN]], <2 x double> [[POWER]]) +; NEON-NEXT: ret <2 x double> [[TMP1]] +; +; SVE-LABEL: define <2 x double> @llvm_pow_f64 +; SVE-SAME: (<2 x double> [[IN:%.*]], <2 x double> [[POWER:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <2 x double> @llvm.pow.v2f64(<2 x double> [[IN]], <2 x double> [[POWER]]) +; SVE-NEXT: ret <2 x double> [[TMP1]] +; + %1 = call fast <2 x double> @llvm.pow.v2f64(<2 x double> %in, <2 x double> %power) + ret <2 x double> %1 +} + +define <4 x float> @llvm_pow_f32(<4 x float> %in, <4 x float> %power) { +; NEON-LABEL: define <4 x float> @llvm_pow_f32 +; NEON-SAME: (<4 x float> [[IN:%.*]], <4 x float> [[POWER:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast <4 x float> @llvm.pow.v4f32(<4 x float> [[IN]], <4 x float> [[POWER]]) +; NEON-NEXT: ret <4 x float> [[TMP1]] +; +; SVE-LABEL: define <4 x float> @llvm_pow_f32 +; SVE-SAME: (<4 x float> [[IN:%.*]], <4 x float> [[POWER:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast <4 x float> @llvm.pow.v4f32(<4 x float> [[IN]], <4 x float> [[POWER]]) +; SVE-NEXT: ret <4 x float> [[TMP1]] +; + %1 = call fast <4 x float> @llvm.pow.v4f32(<4 x float> %in, <4 x float> %power) + ret <4 x float> %1 +} + +define @llvm_pow_vscale_f64( %in, %power) { +; NEON-LABEL: define @llvm_pow_vscale_f64 +; NEON-SAME: ( [[IN:%.*]], [[POWER:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.pow.nxv2f64( [[IN]], [[POWER]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_pow_vscale_f64 +; SVE-SAME: ( [[IN:%.*]], [[POWER:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.pow.nxv2f64( [[IN]], [[POWER]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.pow.nxv2f64( %in, %power) + ret %1 +} + +define @llvm_pow_vscale_f32( %in, %power) { +; NEON-LABEL: define @llvm_pow_vscale_f32 +; NEON-SAME: ( [[IN:%.*]], [[POWER:%.*]]) { +; NEON-NEXT: [[TMP1:%.*]] = call fast @llvm.pow.nxv4f32( [[IN]], [[POWER]]) +; NEON-NEXT: ret [[TMP1]] +; +; SVE-LABEL: define @llvm_pow_vscale_f32 +; SVE-SAME: ( [[IN:%.*]], [[POWER:%.*]]) #[[ATTR1]] { +; SVE-NEXT: [[TMP1:%.*]] = call fast @llvm.pow.nxv4f32( [[IN]], [[POWER]]) +; SVE-NEXT: ret [[TMP1]] +; + %1 = call fast @llvm.pow.nxv4f32( %in, %power) + ret %1 +} Index: llvm/test/Transforms/LoopVectorize/AArch64/armpl-calls-with-ptr.ll =================================================================== --- /dev/null +++ llvm/test/Transforms/LoopVectorize/AArch64/armpl-calls-with-ptr.ll @@ -0,0 +1,297 @@ +; RUN: opt -vector-library=ArmPL -passes=inject-tli-mappings,loop-vectorize -S < %s | FileCheck %s --check-prefixes=CHECK,NEON +; RUN: opt -mattr=+sve -vector-library=ArmPL -passes=inject-tli-mappings,loop-vectorize -S < %s | FileCheck %s --check-prefixes=CHECK,SVE + +target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128" +target triple = "aarch64-unknown-linux-gnu" + + +;Currently loops with calls to functions taking pointers +;can not be vectorised even if the ptr is accessed in a linear way. + +declare double @sincos(double, ptr, ptr) +declare float @sincosf(float, ptr, ptr) + +define void @linear_sincos_f64(ptr nocapture %in.ptr, ptr %out1.ptr, ptr %out2.ptr) { +; CHECK-LABEL: @linear_sincos_f64( +; NEON-NOT: [[TMP5:%.*]] = call <2 x double> @armpl_vsincosq_f64 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svsincos_f64_x +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds double, ptr %out1.ptr, i64 %iv + %out2.gep = getelementptr inbounds double, ptr %out2.ptr, i64 %iv + tail call void @sincos(double %in, ptr %out1.gep, ptr %out2.gep) + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @linear_sincos_f32(ptr nocapture %in.ptr, ptr %out1.ptr, ptr %out2.ptr) { +; CHECK-LABEL: @linear_sincos_f32( +; NEON-NOT: [[TMP5:%.*]] = call <4 x float> @armpl_vsincosq_f32 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svsincos_f32_x +; CHECK: ret void + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds float, ptr %out1.ptr, i64 %iv + %out2.gep = getelementptr inbounds float, ptr %out2.ptr, i64 %iv + tail call void @sincosf(float %in, ptr %out1.gep, ptr %out2.gep) + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define dso_local void @nonlinear_sincos_f64(ptr nocapture %in.ptr, ptr %out1.ptr, ptr %out2.ptr) { +; CHECK-LABEL: @nonlinear_sincos_f64( +; NEON-NOT: [[TMP5:%.*]] = call <2 x double> @armpl_vsincosq_f64 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svsincos_f64_x +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds double, ptr %out1.ptr, i64 %iv + %out2.gep = getelementptr inbounds double, ptr %out2.ptr, i64 %iv + tail call void @sincos(double %in, ptr %out1.gep, ptr %out2.gep) + %iv.next = add nuw nsw i64 %iv, 2 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define dso_local void @nonlinear_sincos_f32(ptr nocapture %in.ptr, ptr %out1.ptr, ptr %out2.ptr) { +; CHECK-LABEL: @nonlinear_sincos_f32( +; NEON-NOT:[[TMP5:%.*]] = call <4 x float> @armpl_vsincosq_f32 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svsincos_f32_x +; CHECK: ret void + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds float, ptr %out1.ptr, i64 %iv + %out2.gep = getelementptr inbounds float, ptr %out2.ptr, i64 %iv + tail call void @sincosf(float %in, ptr %out1.gep, ptr %out2.gep) + %iv.next = add nuw nsw i64 %iv, 2 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @sincospi(double, ptr, ptr) +declare float @sincospif(float, ptr, ptr) + +define void @linear_sincospi_f64(ptr nocapture %in.ptr, ptr %out1.ptr, ptr %out2.ptr) { +; CHECK-LABEL: @linear_sincospi_f64( +; NEON-NOT: [[TMP5:%.*]] = call <2 x double> @armpl_vsincospiq_f64 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svsincospi_f64_x +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds double, ptr %out1.ptr, i64 %iv + %out2.gep = getelementptr inbounds double, ptr %out2.ptr, i64 %iv + tail call void @sincospi(double %in, ptr %out1.gep, ptr %out2.gep) + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @linear_sincospi_f32(ptr nocapture %in.ptr, ptr %out1.ptr, ptr %out2.ptr) { +; CHECK-LABEL: @linear_sincospi_f32( +; NEON-NOT: [[TMP5:%.*]] = call <4 x float> @armpl_vsincospiq_f32 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svsincospi_f32_x +; CHECK: ret void + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds float, ptr %out1.ptr, i64 %iv + %out2.gep = getelementptr inbounds float, ptr %out2.ptr, i64 %iv + tail call void @sincospif(float %in, ptr %out1.gep, ptr %out2.gep) + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define dso_local void @nonlinear_sincospi_f64(ptr nocapture %in.ptr, ptr %out1.ptr, ptr %out2.ptr) { +; CHECK-LABEL: @nonlinear_sincospi_f64( +; NEON-NOT: [[TMP5:%.*]] = call <2 x double> @armpl_vsincospiq_f64 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svsincospi_f64_x +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds double, ptr %out1.ptr, i64 %iv + %out2.gep = getelementptr inbounds double, ptr %out2.ptr, i64 %iv + tail call void @sincospi(double %in, ptr %out1.gep, ptr %out2.gep) + %iv.next = add nuw nsw i64 %iv, 2 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define dso_local void @nonlinear_sincospi_f32(ptr nocapture %in.ptr, ptr %out1.ptr, ptr %out2.ptr) { +; CHECK-LABEL: @nonlinear_sincospi_f32( +; NEON-NOT:[[TMP5:%.*]] = call <4 x float> @armpl_vsincospiq_f32 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svsincospi_f32_x +; CHECK: ret void + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds float, ptr %out1.ptr, i64 %iv + %out2.gep = getelementptr inbounds float, ptr %out2.ptr, i64 %iv + tail call void @sincospif(float %in, ptr %out1.gep, ptr %out2.gep) + %iv.next = add nuw nsw i64 %iv, 2 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @modf(double, ptr) +declare float @modff(float, ptr) + +define void @linear_modf_f64(ptr nocapture %in.ptr, ptr %out1.ptr) { +; CHECK-LABEL: @linear_modf_f64( +; NEON-NOT: [[TMP5:%.*]] = call <2 x double> @armpl_vmodfq_f64 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svmodf_f64_x +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds double, ptr %out1.ptr, i64 %iv + tail call void @modf(double %in, ptr %out1.gep) + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @linear_modf_f32(ptr nocapture %in.ptr, ptr %out1.ptr) { +; CHECK-LABEL: @linear_modf_f32( +; NEON-NOT: [[TMP5:%.*]] = call <4 x float> @armpl_vmodfq_f32 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svmodf_f32_x +; CHECK: ret void + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds float, ptr %out1.ptr, i64 %iv + tail call void @modff(float %in, ptr %out1.gep) + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @nonlinear_modf_f64(ptr nocapture %in.ptr, ptr %out1.ptr) { +; CHECK-LABEL: @nonlinear_modf_f64( +; NEON-NOT: [[TMP5:%.*]] = call <2 x double> @armpl_vmodfq_f64 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svmodf_f64_x +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds double, ptr %out1.ptr, i64 %iv + tail call void @modf(double %in, ptr %out1.gep) + %iv.next = add nuw nsw i64 %iv, 2 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @nonlinear_modf_f32(ptr nocapture %in.ptr, ptr %out1.ptr) { +; CHECK-LABEL: @nonlinear_modf_f32( +; NEON-NOT: [[TMP5:%.*]] = call <4 x float> @armpl_vmodfq_f32 +; SVE-NOT: [[TMP5:%.*]] = call @armpl_svmodf_f32_x +; CHECK: ret void + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %out1.gep = getelementptr inbounds float, ptr %out1.ptr, i64 %iv + tail call void @modff(float %in, ptr %out1.gep) + %iv.next = add nuw nsw i64 %iv, 2 + %exitcond = icmp eq i64 %iv.next, 1024 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + Index: llvm/test/Transforms/LoopVectorize/AArch64/armpl-calls.ll =================================================================== --- /dev/null +++ llvm/test/Transforms/LoopVectorize/AArch64/armpl-calls.ll @@ -0,0 +1,1950 @@ +; RUN: opt -vector-library=ArmPL -passes=inject-tli-mappings,loop-vectorize -S < %s | FileCheck %s --check-prefixes=CHECK,NEON +; RUN: opt -mattr=+sve -vector-library=ArmPL -passes=inject-tli-mappings,loop-vectorize -S < %s | FileCheck %s --check-prefixes=CHECK,SVE + +target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128" +target triple = "aarch64-unknown-linux-gnu" + + +; Tests are checking if LV can vectorize loops with function calls +; using mappings from TLI for scalable and fixed width vectorization. + +declare double @acos(double) +declare float @acosf(float) + +define void @acos_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @acos_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vacosq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svacos_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @acos(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @acos_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @acos_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vacosq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svacos_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @acosf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @acosh(double) +declare float @acoshf(float) + +define void @acosh_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @acosh_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vacoshq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svacosh_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @acosh(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @acosh_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @acosh_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vacoshq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svacosh_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @acoshf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @asin(double) +declare float @asinf(float) + +define void @asin_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @asin_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vasinq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svasin_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @asin(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @asin_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @asin_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vasinq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svasin_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @asinf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @asinh(double) +declare float @asinhf(float) + +define void @asinh_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @asinh_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vasinhq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svasinh_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @asinh(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @asinh_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @asinh_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vasinhq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svasinh_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @asinhf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @atan(double) +declare float @atanf(float) + +define void @atan_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @atan_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vatanq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svatan_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @atan(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @atan_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @atan_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vatanq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svatan_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @atanf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @atanh(double) +declare float @atanhf(float) + +define void @atanh_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @atanh_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vatanhq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svatanh_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @atanh(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @atanh_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @atanh_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vatanhq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svatanh_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @atanhf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @cbrt(double) +declare float @cbrtf(float) + +define void @cbrt_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @cbrt_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vcbrtq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svcbrt_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @cbrt(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @cbrt_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @cbrt_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vcbrtq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svcbrt_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @cbrtf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @cos(double) +declare float @cosf(float) + +define void @cos_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @cos_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vcosq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svcos_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @cos(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @cos_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @cos_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vcosq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svcos_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @cosf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @cosh(double) +declare float @coshf(float) + +define void @cosh_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @cosh_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vcoshq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svcosh_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @cosh(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @cosh_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @cosh_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vcoshq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svcosh_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @coshf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @erf(double) +declare float @erff(float) + +define void @erf_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @erf_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_verfq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_sverf_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @erf(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @erf_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @erf_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_verfq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_sverf_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @erff(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @erfc(double) +declare float @erfcf(float) + +define void @erfc_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @erfc_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_verfcq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_sverfc_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @erfc(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @erfc_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @erfc_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_verfcq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_sverfc_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @erfcf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @exp(double) +declare float @expf(float) + +define void @exp_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @exp_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vexpq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexp_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @exp(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @exp_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @exp_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vexpq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexp_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @expf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @exp2(double) +declare float @exp2f(float) + +define void @exp2_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @exp2_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vexp2q_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexp2_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @exp2(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @exp2_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @exp2_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vexp2q_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexp2_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @exp2f(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @exp10(double) +declare float @exp10f(float) + +define void @exp10_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @exp10_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vexp10q_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexp10_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @exp10(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @exp10_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @exp10_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vexp10q_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexp10_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @exp10f(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @expm1(double) +declare float @expm1f(float) + +define void @expm1_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @expm1_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vexpm1q_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexpm1_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @expm1(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @expm1_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @expm1_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vexpm1q_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexpm1_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @expm1f(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @lgamma(double) +declare float @lgammaf(float) + +define void @lgamma_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @lgamma_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vlgammaq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlgamma_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @lgamma(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @lgamma_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @lgamma_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vlgammaq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlgamma_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @lgammaf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @log(double) +declare float @logf(float) + +define void @log_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vlogq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @log(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @log_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vlogq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @logf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @log1p(double) +declare float @log1pf(float) + +define void @log1p_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log1p_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vlog1pq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog1p_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @log1p(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @log1p_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log1p_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vlog1pq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog1p_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @log1pf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @log2(double) +declare float @log2f(float) + +define void @log2_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log2_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vlog2q_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog2_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @log2(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @log2_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log2_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vlog2q_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog2_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @log2f(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @log10(double) +declare float @log10f(float) + +define void @log10_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log10_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vlog10q_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog10_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @log10(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @log10_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log10_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vlog10q_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog10_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @log10f(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @sin(double) +declare float @sinf(float) + +define void @sin_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @sin_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vsinq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svsin_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @sin(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @sin_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @sin_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vsinq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svsin_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @sinf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @sinh(double) +declare float @sinhf(float) + +define void @sinh_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @sinh_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vsinhq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svsinh_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @sinh(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @sinh_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @sinh_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vsinhq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svsinh_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @sinhf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @sinpi(double) +declare float @sinpif(float) + +define void @sinpi_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @sinpi_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vsinpiq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svsinpi_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @sinpi(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @sinpi_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @sinpi_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vsinpiq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svsinpi_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @sinpif(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @sqrt(double) +declare float @sqrtf(float) + +define void @sqrt_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @sqrt_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vsqrtq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svsqrt_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @sqrt(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @sqrt_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @sqrt_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vsqrtq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svsqrt_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @sqrtf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @tan(double) +declare float @tanf(float) + +define void @tan_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @tan_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vtanq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svtan_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @tan(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @tan_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @tan_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vtanq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svtan_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @tanf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @tanh(double) +declare float @tanhf(float) + +define void @tanh_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @tanh_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vtanhq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svtanh_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @tanh(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @tanh_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @tanh_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vtanhq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svtanh_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @tanhf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @tgamma(double) +declare float @tgammaf(float) + +define void @tgamma_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @tgamma_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vtgammaq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svtgamma_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @tgamma(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @tgamma_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @tgamma_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vtgammaq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svtgamma_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @tgammaf(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @atan2(double, double) +declare float @atan2f(float, float) + +define void @atan2_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @atan2_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vatan2q_f64(<2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svatan2_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @atan2(double %in, double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @atan2_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @atan2_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vatan2q_f32(<4 x float> [[TMP4:%.*]], <4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svatan2_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @atan2f(float %in, float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @copysign(double, double) +declare float @copysignf(float, float) + +define void @copysign_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @copysign_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vcopysignq_f64(<2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svcopysign_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @copysign(double %in, double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @copysign_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @copysign_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vcopysignq_f32(<4 x float> [[TMP4:%.*]], <4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svcopysign_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @copysignf(float %in, float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @fdim(double, double) +declare float @fdimf(float, float) + +define void @fdim_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @fdim_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vfdimq_f64(<2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svfdim_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @fdim(double %in, double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @fdim_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @fdim_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vfdimq_f32(<4 x float> [[TMP4:%.*]], <4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svfdim_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @fdimf(float %in, float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @fmin(double, double) +declare float @fminf(float, float) + +define void @fmin_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @fmin_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vfminq_f64(<2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svfmin_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @fmin(double %in, double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @fmin_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @fmin_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vfminq_f32(<4 x float> [[TMP4:%.*]], <4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svfmin_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @fminf(float %in, float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @fmod(double, double) +declare float @fmodf(float, float) + +define void @fmod_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @fmod_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vfmodq_f64(<2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svfmod_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @fmod(double %in, double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @fmod_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @fmod_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vfmodq_f32(<4 x float> [[TMP4:%.*]], <4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svfmod_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @fmodf(float %in, float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @hypot(double, double) +declare float @hypotf(float, float) + +define void @hypot_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @hypot_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vhypotq_f64(<2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svhypot_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @hypot(double %in, double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @hypot_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @hypot_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vhypotq_f32(<4 x float> [[TMP4:%.*]], <4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svhypot_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @hypotf(float %in, float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @nextafter(double, double) +declare float @nextafterf(float, float) + +define void @nextafter_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @nextafter_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vnextafterq_f64(<2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svnextafter_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @nextafter(double %in, double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @nextafter_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @nextafter_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vnextafterq_f32(<4 x float> [[TMP4:%.*]], <4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svnextafter_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @nextafterf(float %in, float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @pow(double, double) +declare float @powf(float, float) + +define void @pow_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @pow_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vpowq_f64(<2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svpow_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @pow(double %in, double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @pow_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @pow_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vpowq_f32(<4 x float> [[TMP4:%.*]], <4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svpow_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @powf(float %in, float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @fma(double, double, double) +declare float @fmaf(float, float, float) + +define void @fma_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @fma_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vfmaq_f64(<2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svfma_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @fma(double %in, double %in, double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @fma_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @fma_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vfmaq_f32(<4 x float> [[TMP4:%.*]], <4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svfma_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @fmaf(float %in, float %in, float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare i32 @ilogb(double) +declare i32 @ilogbf(float) + +define void @ilogb_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @ilogb_f64( +; NEON: [[TMP5:%.*]] = call <2 x i32> @armpl_vilogbq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svilogb_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call i32 @ilogb(double %in) + %out.gep = getelementptr inbounds i32, ptr %out.ptr, i64 %iv + store i32 %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @ilogb_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @ilogb_f32( +; NEON: [[TMP5:%.*]] = call <4 x i32> @armpl_vilogbq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svilogb_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call i32 @ilogbf(float %in) + %out.gep = getelementptr inbounds i32, ptr %out.ptr, i64 %iv + store i32 %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @ldexp(double, i32) +declare float @ldexpf(float, i32) + +define void @ldexp_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @ldexp_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vldexpq_f64(<2 x double> [[TMP4:%.*]], <2 x i32> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svldexp_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.int = trunc i64 %iv to i32 + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @ldexp(double %in, i32 %in.int) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @ldexp_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @ldexp_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vldexpq_f32(<4 x float> [[TMP4:%.*]], <4 x i32> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svldexp_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.int = trunc i64 %iv to i32 + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @ldexpf(float %in, i32 %in.int) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + Index: llvm/test/Transforms/LoopVectorize/AArch64/armpl-intrinsics.ll =================================================================== --- /dev/null +++ llvm/test/Transforms/LoopVectorize/AArch64/armpl-intrinsics.ll @@ -0,0 +1,418 @@ +; RUN: opt -vector-library=ArmPL -passes=inject-tli-mappings,loop-vectorize -S < %s | FileCheck %s --check-prefixes=CHECK,NEON +; RUN: opt -mattr=+sve -vector-library=ArmPL -passes=inject-tli-mappings,loop-vectorize -S < %s | FileCheck %s --check-prefixes=CHECK,SVE + +target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128" +target triple = "aarch64-unknown-linux-gnu" + + +; Tests are checking if LV can vectorize loops with llvm math intrinsics +; using mappings from TLI for scalable and fixed width vectorization. + +declare double @llvm.cos.f64(double) +declare float @llvm.cos.f32(float) + +define void @cos_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @cos_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vcosq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svcos_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @llvm.cos.f64(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @cos_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @cos_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vcosq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svcos_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @llvm.cos.f32(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @llvm.exp.f64(double) +declare float @llvm.exp.f32(float) + +define void @exp_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @exp_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vexpq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexp_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @llvm.exp.f64(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @exp_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @exp_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vexpq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexp_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @llvm.exp.f32(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @llvm.exp2.f64(double) +declare float @llvm.exp2.f32(float) + +define void @exp2_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @exp2_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vexp2q_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexp2_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @llvm.exp2.f64(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @exp2_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @exp2_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vexp2q_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svexp2_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @llvm.exp2.f32(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @llvm.log.f64(double) +declare float @llvm.log.f32(float) + +define void @log_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vlogq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @llvm.log.f64(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @log_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vlogq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @llvm.log.f32(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @llvm.log2.f64(double) +declare float @llvm.log2.f32(float) + +define void @log2_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log2_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vlog2q_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog2_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @llvm.log2.f64(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @log2_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log2_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vlog2q_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog2_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @llvm.log2.f32(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @llvm.log10.f64(double) +declare float @llvm.log10.f32(float) + +define void @log10_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log10_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vlog10q_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog10_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @llvm.log10.f64(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @log10_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @log10_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vlog10q_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svlog10_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @llvm.log10.f32(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @llvm.sin.f64(double) +declare float @llvm.sin.f32(float) + +define void @sin_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @sin_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vsinq_f64(<2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svsin_f64_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @llvm.sin.f64(double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @sin_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @sin_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vsinq_f32(<4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svsin_f32_x( [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @llvm.sin.f32(float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +declare double @llvm.pow.f64(double, double) +declare float @llvm.pow.f32(float, float) + +define void @pow_f64(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @pow_f64( +; NEON: [[TMP5:%.*]] = call <2 x double> @armpl_vpowq_f64(<2 x double> [[TMP4:%.*]], <2 x double> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svpow_f64_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds double, ptr %in.ptr, i64 %iv + %in = load double, ptr %in.gep, align 8 + %call = tail call double @llvm.pow.f64(double %in, double %in) + %out.gep = getelementptr inbounds double, ptr %out.ptr, i64 %iv + store double %call, ptr %out.gep, align 8 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} + +define void @pow_f32(ptr nocapture %in.ptr, ptr %out.ptr) { +; CHECK-LABEL: @pow_f32( +; NEON: [[TMP5:%.*]] = call <4 x float> @armpl_vpowq_f32(<4 x float> [[TMP4:%.*]], <4 x float> [[TMP4:%.*]]) +; SVE: [[TMP5:%.*]] = call @armpl_svpow_f32_x( [[TMP4:%.*]], [[TMP4:%.*]], {{.*}}) +; CHECK: ret void +; + entry: + br label %for.body + + for.body: + %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] + %in.gep = getelementptr inbounds float, ptr %in.ptr, i64 %iv + %in = load float, ptr %in.gep, align 8 + %call = tail call float @llvm.pow.f32(float %in, float %in) + %out.gep = getelementptr inbounds float, ptr %out.ptr, i64 %iv + store float %call, ptr %out.gep, align 4 + %iv.next = add nuw nsw i64 %iv, 1 + %exitcond = icmp eq i64 %iv.next, 1000 + br i1 %exitcond, label %for.end, label %for.body + + for.end: + ret void +} +