Yingwei Zheng cab09e76e0
[InstCombine] Propagate FMF from fptrunc when folding fptrunc fabs(X) -> fabs(fptrunc X) (#143352)
Alive2: https://alive2.llvm.org/ce/z/DWV3G3
fptrunc yields infinity when the input cannot fit in the target type. So
ninf should be propagated from fptrunc. For other intrinsics, the
previous check ensures that the result is never an infinity:

5d3899d293/llvm/lib/Transforms/InstCombine/InstCombineCasts.cpp (L1910-L1917)

Closes https://github.com/llvm/llvm-project/issues/143122.
2025-06-16 12:07:47 +08:00

556 lines
16 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; Test some floating point casting cases
; RUN: opt < %s -passes=instcombine -S | FileCheck %s
define i8 @test1() {
; CHECK-LABEL: @test1(
; CHECK-NEXT: ret i8 -1
;
%x = fptoui float 2.550000e+02 to i8
ret i8 %x
}
define i8 @test2() {
; CHECK-LABEL: @test2(
; CHECK-NEXT: ret i8 -1
;
%x = fptosi float -1.000000e+00 to i8
ret i8 %x
}
define half @test3(float %a) {
; CHECK-LABEL: @test3(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = call half @llvm.fabs.f16(half [[TMP1]])
; CHECK-NEXT: ret half [[C]]
;
%b = call float @llvm.fabs.f32(float %a)
%c = fptrunc float %b to half
ret half %c
}
define half @test3_fast(float %a) {
; CHECK-LABEL: @test3_fast(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = call fast half @llvm.fabs.f16(half [[TMP1]])
; CHECK-NEXT: ret half [[C]]
;
%b = call float @llvm.fabs.f32(float %a)
%c = fptrunc fast float %b to half
ret half %c
}
define half @test3_both_ninf(float %a) {
; CHECK-LABEL: @test3_both_ninf(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = call ninf half @llvm.fabs.f16(half [[TMP1]])
; CHECK-NEXT: ret half [[C]]
;
%b = call ninf float @llvm.fabs.f32(float %a)
%c = fptrunc ninf float %b to half
ret half %c
}
define half @test3_fabs_ninf(float %a) {
; CHECK-LABEL: @test3_fabs_ninf(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = call half @llvm.fabs.f16(half [[TMP1]])
; CHECK-NEXT: ret half [[C]]
;
%b = call ninf float @llvm.fabs.f32(float %a)
%c = fptrunc float %b to half
ret half %c
}
define half @test3_fptrunc_ninf(float %a) {
; CHECK-LABEL: @test3_fptrunc_ninf(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = call ninf half @llvm.fabs.f16(half [[TMP1]])
; CHECK-NEXT: ret half [[C]]
;
%b = call float @llvm.fabs.f32(float %a)
%c = fptrunc ninf float %b to half
ret half %c
}
define half @fneg_fptrunc(float %a) {
; CHECK-LABEL: @fneg_fptrunc(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = fneg half [[TMP1]]
; CHECK-NEXT: ret half [[C]]
;
%b = fsub float -0.0, %a
%c = fptrunc float %b to half
ret half %c
}
define half @unary_fneg_fptrunc(float %a) {
; CHECK-LABEL: @unary_fneg_fptrunc(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = fneg half [[TMP1]]
; CHECK-NEXT: ret half [[C]]
;
%b = fneg float %a
%c = fptrunc float %b to half
ret half %c
}
define <2 x half> @fneg_fptrunc_vec_poison(<2 x float> %a) {
; CHECK-LABEL: @fneg_fptrunc_vec_poison(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc <2 x float> [[A:%.*]] to <2 x half>
; CHECK-NEXT: [[C:%.*]] = fneg <2 x half> [[TMP1]]
; CHECK-NEXT: ret <2 x half> [[C]]
;
%b = fsub <2 x float> <float -0.0, float poison>, %a
%c = fptrunc <2 x float> %b to <2 x half>
ret <2 x half> %c
}
define <2 x half> @unary_fneg_fptrunc_vec(<2 x float> %a) {
; CHECK-LABEL: @unary_fneg_fptrunc_vec(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc <2 x float> [[A:%.*]] to <2 x half>
; CHECK-NEXT: [[C:%.*]] = fneg <2 x half> [[TMP1]]
; CHECK-NEXT: ret <2 x half> [[C]]
;
%b = fneg <2 x float> %a
%c = fptrunc <2 x float> %b to <2 x half>
ret <2 x half> %c
}
define half @test4-fast(float %a) {
; CHECK-LABEL: @test4-fast(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc fast float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = fneg fast half [[TMP1]]
; CHECK-NEXT: ret half [[C]]
;
%b = fsub fast float -0.0, %a
%c = fptrunc fast float %b to half
ret half %c
}
define half @test4-mixed-fast-1(float %a) {
; CHECK-LABEL: @test4-mixed-fast-1(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = fneg half [[TMP1]]
; CHECK-NEXT: ret half [[C]]
;
%b = fsub float -0.0, %a
%c = fptrunc fast float %b to half
ret half %c
}
define half @test4-mixed-fast-2(float %a) {
; CHECK-LABEL: @test4-mixed-fast-2(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = fneg half [[TMP1]]
; CHECK-NEXT: ret half [[C]]
;
%b = fsub fast float -0.0, %a
%c = fptrunc float %b to half
ret half %c
}
define half @test4_unary_fneg-fast(float %a) {
; CHECK-LABEL: @test4_unary_fneg-fast(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc fast float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = fneg fast half [[TMP1]]
; CHECK-NEXT: ret half [[C]]
;
%b = fneg fast float %a
%c = fptrunc fast float %b to half
ret half %c
}
define half @test4_unary_fneg-mixed-fast-1(float %a) {
; CHECK-LABEL: @test4_unary_fneg-mixed-fast-1(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = fneg half [[TMP1]]
; CHECK-NEXT: ret half [[C]]
;
%b = fneg float %a
%c = fptrunc fast float %b to half
ret half %c
}
define half @test4_unary_fneg-mixed-fast-2(float %a) {
; CHECK-LABEL: @test4_unary_fneg-mixed-fast-2(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc float [[A:%.*]] to half
; CHECK-NEXT: [[C:%.*]] = fneg half [[TMP1]]
; CHECK-NEXT: ret half [[C]]
;
%b = fneg fast float %a
%c = fptrunc float %b to half
ret half %c
}
define <2 x half> @test4_unary_fneg-vec-fast(<2 x float> %a) {
; CHECK-LABEL: @test4_unary_fneg-vec-fast(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc fast <2 x float> [[A:%.*]] to <2 x half>
; CHECK-NEXT: [[C:%.*]] = fneg fast <2 x half> [[TMP1]]
; CHECK-NEXT: ret <2 x half> [[C]]
;
%b = fneg fast <2 x float> %a
%c = fptrunc fast <2 x float> %b to <2 x half>
ret <2 x half> %c
}
define <2 x half> @test4_unary_fneg-vec-mixed-fast-1(<2 x float> %a) {
; CHECK-LABEL: @test4_unary_fneg-vec-mixed-fast-1(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc <2 x float> [[A:%.*]] to <2 x half>
; CHECK-NEXT: [[C:%.*]] = fneg <2 x half> [[TMP1]]
; CHECK-NEXT: ret <2 x half> [[C]]
;
%b = fneg <2 x float> %a
%c = fptrunc fast <2 x float> %b to <2 x half>
ret <2 x half> %c
}
define <2 x half> @test4_unary_fneg-vec-mixed-fast-2(<2 x float> %a) {
; CHECK-LABEL: @test4_unary_fneg-vec-mixed-fast-2(
; CHECK-NEXT: [[TMP1:%.*]] = fptrunc <2 x float> [[A:%.*]] to <2 x half>
; CHECK-NEXT: [[C:%.*]] = fneg <2 x half> [[TMP1]]
; CHECK-NEXT: ret <2 x half> [[C]]
;
%b = fneg fast <2 x float> %a
%c = fptrunc <2 x float> %b to <2 x half>
ret <2 x half> %c
}
define half @test5(float %a, float %b, float %c) {
; CHECK-LABEL: @test5(
; CHECK-NEXT: [[D:%.*]] = fcmp ogt float [[A:%.*]], [[B:%.*]]
; CHECK-NEXT: [[E:%.*]] = select i1 [[D]], float [[C:%.*]], float 1.000000e+00
; CHECK-NEXT: [[F:%.*]] = fptrunc float [[E]] to half
; CHECK-NEXT: ret half [[F]]
;
%d = fcmp ogt float %a, %b
%e = select i1 %d, float %c, float 1.0
%f = fptrunc float %e to half
ret half %f
}
declare float @llvm.fabs.f32(float) nounwind readonly
define <1 x float> @test6(<1 x double> %V) {
; CHECK-LABEL: @test6(
; CHECK-NEXT: [[FREM:%.*]] = frem <1 x double> [[V:%.*]], [[V]]
; CHECK-NEXT: [[TRUNC:%.*]] = fptrunc <1 x double> [[FREM]] to <1 x float>
; CHECK-NEXT: ret <1 x float> [[TRUNC]]
;
%frem = frem <1 x double> %V, %V
%trunc = fptrunc <1 x double> %frem to <1 x float>
ret <1 x float> %trunc
}
define float @test7(double %V) {
; CHECK-LABEL: @test7(
; CHECK-NEXT: [[FREM:%.*]] = frem double [[V:%.*]], 1.000000e+00
; CHECK-NEXT: [[TRUNC:%.*]] = fptrunc double [[FREM]] to float
; CHECK-NEXT: ret float [[TRUNC]]
;
%frem = frem double %V, 1.000000e+00
%trunc = fptrunc double %frem to float
ret float %trunc
}
define float @test8(float %V) {
; CHECK-LABEL: @test8(
; CHECK-NEXT: [[FEXT:%.*]] = fpext float [[V:%.*]] to double
; CHECK-NEXT: [[FREM:%.*]] = frem double [[FEXT]], 1.000000e-01
; CHECK-NEXT: [[TRUNC:%.*]] = fptrunc double [[FREM]] to float
; CHECK-NEXT: ret float [[TRUNC]]
;
%fext = fpext float %V to double
%frem = frem double %fext, 1.000000e-01
%trunc = fptrunc double %frem to float
ret float %trunc
}
define half @test_fptrunc_fptrunc(double %V) {
; CHECK-LABEL: @test_fptrunc_fptrunc(
; CHECK-NEXT: [[T1:%.*]] = fptrunc double [[V:%.*]] to float
; CHECK-NEXT: [[T2:%.*]] = fptrunc float [[T1]] to half
; CHECK-NEXT: ret half [[T2]]
;
%t1 = fptrunc double %V to float
%t2 = fptrunc float %t1 to half
ret half %t2
}
define half @sint_to_fptrunc(i32 %x) {
; CHECK-LABEL: @sint_to_fptrunc(
; CHECK-NEXT: [[F:%.*]] = sitofp i32 [[X:%.*]] to float
; CHECK-NEXT: [[R:%.*]] = fptrunc float [[F]] to half
; CHECK-NEXT: ret half [[R]]
;
%f = sitofp i32 %x to float
%r = fptrunc float %f to half
ret half %r
}
define half @masked_sint_to_fptrunc1(i32 %x) {
; CHECK-LABEL: @masked_sint_to_fptrunc1(
; CHECK-NEXT: [[M:%.*]] = and i32 [[X:%.*]], 16777215
; CHECK-NEXT: [[R:%.*]] = uitofp nneg i32 [[M]] to half
; CHECK-NEXT: ret half [[R]]
;
%m = and i32 %x, 16777215
%f = sitofp i32 %m to float
%r = fptrunc float %f to half
ret half %r
}
define half @masked_sint_to_fptrunc2(i32 %x) {
; CHECK-LABEL: @masked_sint_to_fptrunc2(
; CHECK-NEXT: [[M:%.*]] = lshr i32 [[X:%.*]], 8
; CHECK-NEXT: [[R:%.*]] = uitofp nneg i32 [[M]] to half
; CHECK-NEXT: ret half [[R]]
;
%m = lshr i32 %x, 8
%f = sitofp i32 %m to float
%r = fptrunc float %f to half
ret half %r
}
define half @masked_sint_to_fptrunc3(i32 %x) {
; CHECK-LABEL: @masked_sint_to_fptrunc3(
; CHECK-NEXT: [[M:%.*]] = lshr i32 [[X:%.*]], 7
; CHECK-NEXT: [[F:%.*]] = uitofp nneg i32 [[M]] to float
; CHECK-NEXT: [[R:%.*]] = fptrunc float [[F]] to half
; CHECK-NEXT: ret half [[R]]
;
%m = lshr i32 %x, 7
%f = sitofp i32 %m to float
%r = fptrunc float %f to half
ret half %r
}
define double @sint_to_fpext(i32 %x) {
; CHECK-LABEL: @sint_to_fpext(
; CHECK-NEXT: [[F:%.*]] = sitofp i32 [[X:%.*]] to float
; CHECK-NEXT: [[R:%.*]] = fpext float [[F]] to double
; CHECK-NEXT: ret double [[R]]
;
%f = sitofp i32 %x to float
%r = fpext float %f to double
ret double %r
}
define double @masked_sint_to_fpext1(i32 %x) {
; CHECK-LABEL: @masked_sint_to_fpext1(
; CHECK-NEXT: [[M:%.*]] = and i32 [[X:%.*]], 16777215
; CHECK-NEXT: [[R:%.*]] = uitofp nneg i32 [[M]] to double
; CHECK-NEXT: ret double [[R]]
;
%m = and i32 %x, 16777215
%f = sitofp i32 %m to float
%r = fpext float %f to double
ret double %r
}
define double @masked_sint_to_fpext2(i32 %x) {
; CHECK-LABEL: @masked_sint_to_fpext2(
; CHECK-NEXT: [[M:%.*]] = lshr i32 [[X:%.*]], 8
; CHECK-NEXT: [[R:%.*]] = uitofp nneg i32 [[M]] to double
; CHECK-NEXT: ret double [[R]]
;
%m = lshr i32 %x, 8
%f = sitofp i32 %m to float
%r = fpext float %f to double
ret double %r
}
define double @masked_sint_to_fpext3(i32 %x) {
; CHECK-LABEL: @masked_sint_to_fpext3(
; CHECK-NEXT: [[M:%.*]] = lshr i32 [[X:%.*]], 7
; CHECK-NEXT: [[F:%.*]] = uitofp nneg i32 [[M]] to float
; CHECK-NEXT: [[R:%.*]] = fpext float [[F]] to double
; CHECK-NEXT: ret double [[R]]
;
%m = lshr i32 %x, 7
%f = sitofp i32 %m to float
%r = fpext float %f to double
ret double %r
}
define half @uint_to_fptrunc(i32 %x) {
; CHECK-LABEL: @uint_to_fptrunc(
; CHECK-NEXT: [[F:%.*]] = uitofp i32 [[X:%.*]] to float
; CHECK-NEXT: [[R:%.*]] = fptrunc float [[F]] to half
; CHECK-NEXT: ret half [[R]]
;
%f = uitofp i32 %x to float
%r = fptrunc float %f to half
ret half %r
}
define half @masked_uint_to_fptrunc1(i32 %x) {
; CHECK-LABEL: @masked_uint_to_fptrunc1(
; CHECK-NEXT: [[M:%.*]] = and i32 [[X:%.*]], 16777215
; CHECK-NEXT: [[R:%.*]] = uitofp nneg i32 [[M]] to half
; CHECK-NEXT: ret half [[R]]
;
%m = and i32 %x, 16777215
%f = uitofp i32 %m to float
%r = fptrunc float %f to half
ret half %r
}
define half @masked_uint_to_fptrunc2(i32 %x) {
; CHECK-LABEL: @masked_uint_to_fptrunc2(
; CHECK-NEXT: [[M:%.*]] = lshr i32 [[X:%.*]], 8
; CHECK-NEXT: [[R:%.*]] = uitofp nneg i32 [[M]] to half
; CHECK-NEXT: ret half [[R]]
;
%m = lshr i32 %x, 8
%f = uitofp i32 %m to float
%r = fptrunc float %f to half
ret half %r
}
define half @masked_uint_to_fptrunc3(i32 %x) {
; CHECK-LABEL: @masked_uint_to_fptrunc3(
; CHECK-NEXT: [[M:%.*]] = lshr i32 [[X:%.*]], 7
; CHECK-NEXT: [[F:%.*]] = uitofp nneg i32 [[M]] to float
; CHECK-NEXT: [[R:%.*]] = fptrunc float [[F]] to half
; CHECK-NEXT: ret half [[R]]
;
%m = lshr i32 %x, 7
%f = uitofp i32 %m to float
%r = fptrunc float %f to half
ret half %r
}
define double @uint_to_fpext(i32 %x) {
; CHECK-LABEL: @uint_to_fpext(
; CHECK-NEXT: [[F:%.*]] = uitofp i32 [[X:%.*]] to float
; CHECK-NEXT: [[R:%.*]] = fpext float [[F]] to double
; CHECK-NEXT: ret double [[R]]
;
%f = uitofp i32 %x to float
%r = fpext float %f to double
ret double %r
}
define double @masked_uint_to_fpext1(i32 %x) {
; CHECK-LABEL: @masked_uint_to_fpext1(
; CHECK-NEXT: [[M:%.*]] = and i32 [[X:%.*]], 16777215
; CHECK-NEXT: [[R:%.*]] = uitofp nneg i32 [[M]] to double
; CHECK-NEXT: ret double [[R]]
;
%m = and i32 %x, 16777215
%f = uitofp i32 %m to float
%r = fpext float %f to double
ret double %r
}
define double @masked_uint_to_fpext2(i32 %x) {
; CHECK-LABEL: @masked_uint_to_fpext2(
; CHECK-NEXT: [[M:%.*]] = lshr i32 [[X:%.*]], 8
; CHECK-NEXT: [[R:%.*]] = uitofp nneg i32 [[M]] to double
; CHECK-NEXT: ret double [[R]]
;
%m = lshr i32 %x, 8
%f = uitofp i32 %m to float
%r = fpext float %f to double
ret double %r
}
define double @masked_uint_to_fpext3(i32 %x) {
; CHECK-LABEL: @masked_uint_to_fpext3(
; CHECK-NEXT: [[M:%.*]] = lshr i32 [[X:%.*]], 7
; CHECK-NEXT: [[F:%.*]] = uitofp nneg i32 [[M]] to float
; CHECK-NEXT: [[R:%.*]] = fpext float [[F]] to double
; CHECK-NEXT: ret double [[R]]
;
%m = lshr i32 %x, 7
%f = uitofp i32 %m to float
%r = fpext float %f to double
ret double %r
}
define i32 @fptosi_nonnorm(float nofpclass(norm) %x) {
; CHECK-LABEL: @fptosi_nonnorm(
; CHECK-NEXT: ret i32 0
;
%ret = fptosi float %x to i32
ret i32 %ret
}
define i32 @fptoui_nonnorm(float nofpclass(pnorm) %x) {
; CHECK-LABEL: @fptoui_nonnorm(
; CHECK-NEXT: ret i32 0
;
%ret = fptoui float %x to i32
ret i32 %ret
}
define i32 @fptosi_nonnnorm(float nofpclass(nnorm) %x) {
; CHECK-LABEL: @fptosi_nonnnorm(
; CHECK-NEXT: [[RET:%.*]] = fptosi float [[X:%.*]] to i32
; CHECK-NEXT: ret i32 [[RET]]
;
%ret = fptosi float %x to i32
ret i32 %ret
}
define i32 @fptoui_nonnnorm(float nofpclass(nnorm) %x) {
; CHECK-LABEL: @fptoui_nonnnorm(
; CHECK-NEXT: [[RET:%.*]] = fptoui float [[X:%.*]] to i32
; CHECK-NEXT: ret i32 [[RET]]
;
%ret = fptoui float %x to i32
ret i32 %ret
}
define i32 @fptosi_nonnorm_copysign(float %x) {
; CHECK-LABEL: @fptosi_nonnorm_copysign(
; CHECK-NEXT: ret i32 0
;
%val = call float @llvm.copysign.f32(float 0.0, float %x)
%ret = fptosi float %val to i32
ret i32 %ret
}
define <2 x i32> @fptosi_nonnorm_copysign_vec(<2 x float> %x) {
; CHECK-LABEL: @fptosi_nonnorm_copysign_vec(
; CHECK-NEXT: ret <2 x i32> zeroinitializer
;
%val = call <2 x float> @llvm.copysign.v2f32(<2 x float> zeroinitializer, <2 x float> %x)
%ret = fptosi <2 x float> %val to <2 x i32>
ret <2 x i32> %ret
}
define i32 @fptosi_nonnorm_fmul(float %x) {
; CHECK-LABEL: @fptosi_nonnorm_fmul(
; CHECK-NEXT: [[SEL:%.*]] = fmul float [[X:%.*]], 0.000000e+00
; CHECK-NEXT: [[RET:%.*]] = fptosi float [[SEL]] to i32
; CHECK-NEXT: ret i32 [[RET]]
;
%sel = fmul float %x, 0.000000e+00
%ret = fptosi float %sel to i32
ret i32 %ret
}
define i32 @fptosi_select(i1 %cond) {
; CHECK-LABEL: @fptosi_select(
; CHECK-NEXT: [[RET:%.*]] = select i1 [[COND:%.*]], i32 1, i32 -1
; CHECK-NEXT: ret i32 [[RET]]
;
%sel = select i1 %cond, float 1.0, float -1.0
%ret = fptosi float %sel to i32
ret i32 %ret
}
define i32 @mul_pos_zero_convert(i32 %a) {
; CHECK-LABEL: @mul_pos_zero_convert(
; CHECK-NEXT: entry:
; CHECK-NEXT: ret i32 0
;
entry:
%fp = sitofp i32 %a to float
%ret = fmul float %fp, 0.000000e+00
%conv = fptosi float %ret to i32
ret i32 %conv
}