
This addresses a TODO in foldShuffledIntrinsicOperands to use isTriviallyVectorizable instead of a hardcoded list of intrinsics, which in turn allows more intriniscs to be scalarized by VectorCombine. From what I can tell every intrinsic here should be speculatable so an assertion was added. Because this enables intrinsics like abs which have a scalar operand, we need to also check isVectorIntrinsicWithScalarOpAtArg.
226 lines
7.8 KiB
LLVM
226 lines
7.8 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
|
|
; RUN: opt -S -passes=instcombine < %s | FileCheck %s
|
|
|
|
define float @test1(float %x) nounwind readnone ssp {
|
|
; CHECK-LABEL: @test1(
|
|
; CHECK-NEXT: [[SQRTF:%.*]] = call float @sqrtf(float [[X:%.*]]) #[[ATTR4:[0-9]+]]
|
|
; CHECK-NEXT: ret float [[SQRTF]]
|
|
;
|
|
%conv = fpext float %x to double
|
|
%call = tail call double @sqrt(double %conv) readnone nounwind
|
|
%conv1 = fptrunc double %call to float
|
|
ret float %conv1
|
|
}
|
|
|
|
; PR8096
|
|
|
|
define float @test2(float %x) nounwind readnone ssp {
|
|
; CHECK-LABEL: @test2(
|
|
; CHECK-NEXT: [[SQRTF:%.*]] = call float @sqrtf(float [[X:%.*]]) #[[ATTR4]]
|
|
; CHECK-NEXT: ret float [[SQRTF]]
|
|
;
|
|
%conv = fpext float %x to double
|
|
%call = tail call double @sqrt(double %conv) nounwind
|
|
%conv1 = fptrunc double %call to float
|
|
ret float %conv1
|
|
}
|
|
|
|
; rdar://9763193
|
|
; Can't fold (fptrunc (sqrt (fpext x))) -> (sqrtf x) since there is another
|
|
; use of sqrt result.
|
|
|
|
define float @test3(ptr %v) nounwind uwtable ssp {
|
|
; CHECK-LABEL: @test3(
|
|
; CHECK-NEXT: [[CALL34:%.*]] = call double @sqrt(double 0x7FF8000000000000) #[[ATTR4]]
|
|
; CHECK-NEXT: [[CALL36:%.*]] = call i32 @foo(double [[CALL34]]) #[[ATTR5:[0-9]+]]
|
|
; CHECK-NEXT: [[CONV38:%.*]] = fptrunc double [[CALL34]] to float
|
|
; CHECK-NEXT: ret float [[CONV38]]
|
|
;
|
|
%arrayidx13 = getelementptr inbounds float, ptr %v, i64 2
|
|
%tmp14 = load float, ptr %arrayidx13
|
|
%mul18 = fmul float %tmp14, %tmp14
|
|
%add19 = fadd float undef, %mul18
|
|
%conv = fpext float %add19 to double
|
|
%call34 = call double @sqrt(double %conv) readnone
|
|
%call36 = call i32 (double) @foo(double %call34) nounwind
|
|
%conv38 = fptrunc double %call34 to float
|
|
ret float %conv38
|
|
}
|
|
|
|
; PR43347 - https://bugs.llvm.org/show_bug.cgi?id=43347
|
|
|
|
define void @0(float %f) {
|
|
; CHECK-LABEL: @0(
|
|
; CHECK-NEXT: [[SQRTF:%.*]] = call float @sqrtf(float [[F:%.*]]) #[[ATTR2:[0-9]+]]
|
|
; CHECK-NEXT: ret void
|
|
;
|
|
%d = fpext float %f to double
|
|
%r = call double @sqrt(double %d)
|
|
ret void
|
|
}
|
|
|
|
define float @sqrt_call_nnan_f32(float %x) {
|
|
; CHECK-LABEL: @sqrt_call_nnan_f32(
|
|
; CHECK-NEXT: [[SQRT:%.*]] = call nnan float @sqrtf(float [[X:%.*]])
|
|
; CHECK-NEXT: ret float [[SQRT]]
|
|
;
|
|
%sqrt = call nnan float @sqrtf(float %x)
|
|
ret float %sqrt
|
|
}
|
|
|
|
define double @sqrt_call_nnan_f64(double %x) {
|
|
; CHECK-LABEL: @sqrt_call_nnan_f64(
|
|
; CHECK-NEXT: [[SQRT:%.*]] = tail call nnan ninf double @sqrt(double [[X:%.*]])
|
|
; CHECK-NEXT: ret double [[SQRT]]
|
|
;
|
|
%sqrt = tail call nnan ninf double @sqrt(double %x)
|
|
ret double %sqrt
|
|
}
|
|
|
|
define float @sqrt_call_fabs_f32(float %x) {
|
|
; CHECK-LABEL: @sqrt_call_fabs_f32(
|
|
; CHECK-NEXT: [[A:%.*]] = call float @llvm.fabs.f32(float [[X:%.*]])
|
|
; CHECK-NEXT: [[SQRT:%.*]] = tail call float @sqrtf(float [[A]])
|
|
; CHECK-NEXT: ret float [[SQRT]]
|
|
;
|
|
%a = call float @llvm.fabs.f32(float %x)
|
|
%sqrt = tail call float @sqrtf(float %a)
|
|
ret float %sqrt
|
|
}
|
|
|
|
define double @sqrt_exp(double %x) {
|
|
; CHECK-LABEL: @sqrt_exp(
|
|
; CHECK-NEXT: [[MERGED_SQRT:%.*]] = fmul reassoc double [[X:%.*]], 5.000000e-01
|
|
; CHECK-NEXT: [[E:%.*]] = call reassoc double @llvm.exp.f64(double [[MERGED_SQRT]])
|
|
; CHECK-NEXT: ret double [[E]]
|
|
;
|
|
%e = call reassoc double @llvm.exp.f64(double %x)
|
|
%res = call reassoc double @llvm.sqrt.f64(double %e)
|
|
ret double %res
|
|
}
|
|
|
|
define double @sqrt_exp_2(double %x) {
|
|
; CHECK-LABEL: @sqrt_exp_2(
|
|
; CHECK-NEXT: [[MERGED_SQRT:%.*]] = fmul reassoc double [[X:%.*]], 5.000000e-01
|
|
; CHECK-NEXT: [[E:%.*]] = call reassoc double @exp(double [[MERGED_SQRT]])
|
|
; CHECK-NEXT: ret double [[E]]
|
|
;
|
|
%e = call reassoc double @exp(double %x)
|
|
%res = call reassoc double @sqrt(double %e)
|
|
ret double %res
|
|
}
|
|
|
|
define double @sqrt_exp2(double %x) {
|
|
; CHECK-LABEL: @sqrt_exp2(
|
|
; CHECK-NEXT: [[MERGED_SQRT:%.*]] = fmul reassoc double [[X:%.*]], 5.000000e-01
|
|
; CHECK-NEXT: [[E:%.*]] = call reassoc double @exp2(double [[MERGED_SQRT]])
|
|
; CHECK-NEXT: ret double [[E]]
|
|
;
|
|
%e = call reassoc double @exp2(double %x)
|
|
%res = call reassoc double @sqrt(double %e)
|
|
ret double %res
|
|
}
|
|
|
|
define double @sqrt_exp10(double %x) {
|
|
; CHECK-LABEL: @sqrt_exp10(
|
|
; CHECK-NEXT: [[MERGED_SQRT:%.*]] = fmul reassoc double [[X:%.*]], 5.000000e-01
|
|
; CHECK-NEXT: [[E:%.*]] = call reassoc double @exp10(double [[MERGED_SQRT]])
|
|
; CHECK-NEXT: ret double [[E]]
|
|
;
|
|
%e = call reassoc double @exp10(double %x)
|
|
%res = call reassoc double @sqrt(double %e)
|
|
ret double %res
|
|
}
|
|
|
|
; Negative test
|
|
define double @sqrt_exp_nofast_1(double %x) {
|
|
; CHECK-LABEL: @sqrt_exp_nofast_1(
|
|
; CHECK-NEXT: [[E:%.*]] = call double @llvm.exp.f64(double [[X:%.*]])
|
|
; CHECK-NEXT: [[RES:%.*]] = call reassoc double @llvm.sqrt.f64(double [[E]])
|
|
; CHECK-NEXT: ret double [[RES]]
|
|
;
|
|
%e = call double @llvm.exp.f64(double %x)
|
|
%res = call reassoc double @llvm.sqrt.f64(double %e)
|
|
ret double %res
|
|
}
|
|
|
|
; Negative test
|
|
define double @sqrt_exp_nofast_2(double %x) {
|
|
; CHECK-LABEL: @sqrt_exp_nofast_2(
|
|
; CHECK-NEXT: [[E:%.*]] = call reassoc double @llvm.exp.f64(double [[X:%.*]])
|
|
; CHECK-NEXT: [[RES:%.*]] = call double @llvm.sqrt.f64(double [[E]])
|
|
; CHECK-NEXT: ret double [[RES]]
|
|
;
|
|
%e = call reassoc double @llvm.exp.f64(double %x)
|
|
%res = call double @llvm.sqrt.f64(double %e)
|
|
ret double %res
|
|
}
|
|
|
|
define double @sqrt_exp_merge_constant(double %x, double %y) {
|
|
; CHECK-LABEL: @sqrt_exp_merge_constant(
|
|
; CHECK-NEXT: [[MERGED_SQRT:%.*]] = fmul reassoc nsz double [[X:%.*]], 5.000000e+00
|
|
; CHECK-NEXT: [[E:%.*]] = call reassoc double @llvm.exp.f64(double [[MERGED_SQRT]])
|
|
; CHECK-NEXT: ret double [[E]]
|
|
;
|
|
%mul = fmul reassoc nsz double %x, 10.0
|
|
%e = call reassoc double @llvm.exp.f64(double %mul)
|
|
%res = call reassoc nsz double @llvm.sqrt.f64(double %e)
|
|
ret double %res
|
|
}
|
|
|
|
define double @sqrt_exp_intr_and_libcall(double %x) {
|
|
; CHECK-LABEL: @sqrt_exp_intr_and_libcall(
|
|
; CHECK-NEXT: [[MERGED_SQRT:%.*]] = fmul reassoc double [[X:%.*]], 5.000000e-01
|
|
; CHECK-NEXT: [[E:%.*]] = call reassoc double @exp(double [[MERGED_SQRT]])
|
|
; CHECK-NEXT: ret double [[E]]
|
|
;
|
|
%e = call reassoc double @exp(double %x)
|
|
%res = call reassoc double @llvm.sqrt.f64(double %e)
|
|
ret double %res
|
|
}
|
|
|
|
define double @sqrt_exp_intr_and_libcall_2(double %x) {
|
|
; CHECK-LABEL: @sqrt_exp_intr_and_libcall_2(
|
|
; CHECK-NEXT: [[MERGED_SQRT:%.*]] = fmul reassoc double [[X:%.*]], 5.000000e-01
|
|
; CHECK-NEXT: [[E:%.*]] = call reassoc double @llvm.exp.f64(double [[MERGED_SQRT]])
|
|
; CHECK-NEXT: ret double [[E]]
|
|
;
|
|
%e = call reassoc double @llvm.exp.f64(double %x)
|
|
%res = call reassoc double @sqrt(double %e)
|
|
ret double %res
|
|
}
|
|
|
|
define <2 x float> @sqrt_exp_vec(<2 x float> %x) {
|
|
; CHECK-LABEL: @sqrt_exp_vec(
|
|
; CHECK-NEXT: [[MERGED_SQRT:%.*]] = fmul reassoc <2 x float> [[X:%.*]], splat (float 5.000000e-01)
|
|
; CHECK-NEXT: [[E:%.*]] = call reassoc <2 x float> @llvm.exp.v2f32(<2 x float> [[MERGED_SQRT]])
|
|
; CHECK-NEXT: ret <2 x float> [[E]]
|
|
;
|
|
%e = call reassoc <2 x float> @llvm.exp.v2f32(<2 x float> %x)
|
|
%res = call reassoc <2 x float> @llvm.sqrt.v2f32(<2 x float> %e)
|
|
ret <2 x float> %res
|
|
}
|
|
|
|
define <2 x float> @sqrt_unary_shuffle_ops(<2 x float> %x) {
|
|
; CHECK-LABEL: @sqrt_unary_shuffle_ops(
|
|
; CHECK-NEXT: [[R:%.*]] = call <2 x float> @llvm.sqrt.v2f32(<2 x float> [[A:%.*]])
|
|
; CHECK-NEXT: [[R1:%.*]] = shufflevector <2 x float> [[R]], <2 x float> poison, <2 x i32> <i32 1, i32 0>
|
|
; CHECK-NEXT: ret <2 x float> [[R1]]
|
|
;
|
|
%a = shufflevector <2 x float> %x, <2 x float> poison, <2 x i32> <i32 1, i32 0>
|
|
%r = call <2 x float> @llvm.sqrt(<2 x float> %a)
|
|
ret <2 x float> %r
|
|
}
|
|
|
|
declare i32 @foo(double)
|
|
declare double @sqrt(double) readnone
|
|
declare float @sqrtf(float)
|
|
declare float @llvm.fabs.f32(float)
|
|
declare double @llvm.exp.f64(double)
|
|
declare double @llvm.sqrt.f64(double)
|
|
declare double @exp(double)
|
|
declare double @exp2(double)
|
|
declare double @exp10(double)
|
|
declare <2 x float> @llvm.exp.v2f32(<2 x float>)
|
|
declare <2 x float> @llvm.sqrt.v2f32(<2 x float>)
|