InstCombine: Add baseline tests for copysign with fneg/fabs

diff --git a/llvm/test/Transforms/InstCombine/copysign-fneg-fabs.ll b/llvm/test/Transforms/InstCombine/copysign-fneg-fabs.ll
new file mode 100644 (file)
index 0000000..31846c5
--- /dev/null
+++ b/llvm/test/Transforms/InstCombine/copysign-fneg-fabs.ll
@@ -0,0 +1,292 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt -S -passes=instcombine < %s | FileCheck %s
+
+define half @copysign_fneg_x(half %x, half %y) {
+; CHECK-LABEL: @copysign_fneg_x(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    ret half [[COPYSIGN]]
+;
+  %fneg.x = fneg half %x
+  %copysign = call half @llvm.copysign.f16(half %fneg.x, half %y)
+  ret half %copysign
+}
+
+define half @copysign_fabs_x(half %x, half %y) {
+; CHECK-LABEL: @copysign_fabs_x(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    ret half [[COPYSIGN]]
+;
+  %fabs.x = call half @llvm.fabs.f16(half %x)
+  %copysign = call half @llvm.copysign.f16(half %fabs.x, half %y)
+  ret half %copysign
+}
+
+define half @copysign_fneg_fabs_x(half %x, half %y) {
+; CHECK-LABEL: @copysign_fneg_fabs_x(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    ret half [[COPYSIGN]]
+;
+  %fabs.x = call half @llvm.fabs.f16(half %x)
+  %fneg.fabs.x = fneg half %fabs.x
+  %copysign = call half @llvm.copysign.f16(half %fneg.fabs.x, half %y)
+  ret half %copysign
+}
+
+define half @copysign_fneg_y(half %x, half %y) {
+; CHECK-LABEL: @copysign_fneg_y(
+; CHECK-NEXT:    [[FNEG_Y:%.*]] = fneg half [[Y:%.*]]
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[FNEG_Y]])
+; CHECK-NEXT:    ret half [[COPYSIGN]]
+;
+  %fneg.y = fneg half %y
+  %copysign = call half @llvm.copysign.f16(half %x, half %fneg.y)
+  ret half %copysign
+}
+
+define half @copysign_fabs_y(half %x, half %y) {
+; CHECK-LABEL: @copysign_fabs_y(
+; CHECK-NEXT:    [[TMP1:%.*]] = call half @llvm.fabs.f16(half [[X:%.*]])
+; CHECK-NEXT:    ret half [[TMP1]]
+;
+  %fabs.y = call half @llvm.fabs.f16(half %y)
+  %copysign = call half @llvm.copysign.f16(half %x, half %fabs.y)
+  ret half %copysign
+}
+
+define half @copysign_fneg_fabs_y(half %x, half %y) {
+; CHECK-LABEL: @copysign_fneg_fabs_y(
+; CHECK-NEXT:    [[FABS_Y:%.*]] = call half @llvm.fabs.f16(half [[Y:%.*]])
+; CHECK-NEXT:    [[FNEG_FABS_Y:%.*]] = fneg half [[FABS_Y]]
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[FNEG_FABS_Y]])
+; CHECK-NEXT:    ret half [[COPYSIGN]]
+;
+  %fabs.y = call half @llvm.fabs.f16(half %y)
+  %fneg.fabs.y = fneg half %fabs.y
+  %copysign = call half @llvm.copysign.f16(half %x, half %fneg.fabs.y)
+  ret half %copysign
+}
+
+define half @fneg_copysign(half %x, half %y) {
+; CHECK-LABEL: @fneg_copysign(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FNEG_COPYSIGN:%.*]] = fneg half [[COPYSIGN]]
+; CHECK-NEXT:    ret half [[FNEG_COPYSIGN]]
+;
+  %copysign = call half @llvm.copysign.f16(half %x, half %y)
+  %fneg.copysign = fneg half %copysign
+  ret half %fneg.copysign
+}
+
+define half @fneg_fabs_copysign(half %x, half %y) {
+; CHECK-LABEL: @fneg_fabs_copysign(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    [[FNEG_FABS_COPYSIGN:%.*]] = fneg half [[FABS_COPYSIGN]]
+; CHECK-NEXT:    ret half [[FNEG_FABS_COPYSIGN]]
+;
+  %copysign = call half @llvm.copysign.f16(half %x, half %y)
+  %fabs.copysign = call half @llvm.fabs.f16(half %copysign)
+  %fneg.fabs.copysign = fneg half %fabs.copysign
+  ret half %fneg.fabs.copysign
+}
+
+; https://alive2.llvm.org/ce/z/Ft-7ea
+define half @fabs_copysign(half %x, half %y) {
+; CHECK-LABEL: @fabs_copysign(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    ret half [[FABS_COPYSIGN]]
+;
+  %copysign = call half @llvm.copysign.f16(half %x, half %y)
+  %fabs.copysign = call half @llvm.fabs.f16(half %copysign)
+  ret half %fabs.copysign
+}
+
+define <2 x half> @fneg_copysign_vector(<2 x half> %x, <2 x half> %y) {
+; CHECK-LABEL: @fneg_copysign_vector(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call <2 x half> @llvm.copysign.v2f16(<2 x half> [[X:%.*]], <2 x half> [[Y:%.*]])
+; CHECK-NEXT:    [[FNEG_COPYSIGN:%.*]] = fneg <2 x half> [[COPYSIGN]]
+; CHECK-NEXT:    ret <2 x half> [[FNEG_COPYSIGN]]
+;
+  %copysign = call <2 x half> @llvm.copysign.v2f16(<2 x half> %x, <2 x half> %y)
+  %fneg.copysign = fneg <2 x half> %copysign
+  ret <2 x half> %fneg.copysign
+}
+
+define <2 x half> @fneg_fabs_copysign_vector(<2 x half> %x, <2 x half> %y) {
+; CHECK-LABEL: @fneg_fabs_copysign_vector(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call <2 x half> @llvm.copysign.v2f16(<2 x half> [[X:%.*]], <2 x half> [[Y:%.*]])
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call <2 x half> @llvm.fabs.v2f16(<2 x half> [[COPYSIGN]])
+; CHECK-NEXT:    [[FNEG_FABS_COPYSIGN:%.*]] = fneg <2 x half> [[FABS_COPYSIGN]]
+; CHECK-NEXT:    ret <2 x half> [[FNEG_FABS_COPYSIGN]]
+;
+  %copysign = call <2 x half> @llvm.copysign.v2f16(<2 x half> %x, <2 x half> %y)
+  %fabs.copysign = call <2 x half> @llvm.fabs.v2f16(<2 x half> %copysign)
+  %fneg.fabs.copysign = fneg <2 x half> %fabs.copysign
+  ret <2 x half> %fneg.fabs.copysign
+}
+
+define <2 x half> @fabs_copysign_vector(<2 x half> %x, <2 x half> %y) {
+; CHECK-LABEL: @fabs_copysign_vector(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call <2 x half> @llvm.copysign.v2f16(<2 x half> [[X:%.*]], <2 x half> [[Y:%.*]])
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call <2 x half> @llvm.fabs.v2f16(<2 x half> [[COPYSIGN]])
+; CHECK-NEXT:    ret <2 x half> [[FABS_COPYSIGN]]
+;
+  %copysign = call <2 x half> @llvm.copysign.v2f16(<2 x half> %x, <2 x half> %y)
+  %fabs.copysign = call <2 x half> @llvm.fabs.v2f16(<2 x half> %copysign)
+  ret <2 x half> %fabs.copysign
+}
+
+define half @fneg_copysign_flags(half %x, half %y) {
+; CHECK-LABEL: @fneg_copysign_flags(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call nnan nsz half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FNEG_COPYSIGN:%.*]] = fneg ninf nsz half [[COPYSIGN]]
+; CHECK-NEXT:    ret half [[FNEG_COPYSIGN]]
+;
+  %copysign = call nnan nsz half @llvm.copysign.f16(half %x, half %y)
+  %fneg.copysign = fneg ninf nsz half %copysign
+  ret half %fneg.copysign
+}
+
+define half @fneg_fabs_copysign_flags(half %x, half %y) {
+; CHECK-LABEL: @fneg_fabs_copysign_flags(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call nnan ninf nsz half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call nnan ninf afn half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    [[FNEG_FABS_COPYSIGN:%.*]] = fneg reassoc ninf half [[FABS_COPYSIGN]]
+; CHECK-NEXT:    ret half [[FNEG_FABS_COPYSIGN]]
+;
+  %copysign = call nnan nsz ninf half @llvm.copysign.f16(half %x, half %y)
+  %fabs.copysign = call nnan afn ninf half @llvm.fabs.f16(half %copysign)
+  %fneg.fabs.copysign = fneg reassoc ninf half %fabs.copysign
+  ret half %fneg.fabs.copysign
+}
+
+; Make sure we don't break things by polluting copysign with nsz
+define half @fneg_nsz_copysign(half %x, half %y) {
+; CHECK-LABEL: @fneg_nsz_copysign(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FNEG_COPYSIGN:%.*]] = fneg nsz half [[COPYSIGN]]
+; CHECK-NEXT:    ret half [[FNEG_COPYSIGN]]
+;
+  %copysign = call half @llvm.copysign.f16(half %x, half %y)
+  %fneg.copysign = fneg nsz half %copysign
+  ret half %fneg.copysign
+}
+
+define half @fneg_fabs_copysign_flags_none_fabs(half %x, half %y) {
+; CHECK-LABEL: @fneg_fabs_copysign_flags_none_fabs(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call nnan ninf nsz half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    [[FNEG_FABS_COPYSIGN:%.*]] = fneg fast half [[FABS_COPYSIGN]]
+; CHECK-NEXT:    ret half [[FNEG_FABS_COPYSIGN]]
+;
+  %copysign = call nnan nsz ninf half @llvm.copysign.f16(half %x, half %y)
+  %fabs.copysign = call half @llvm.fabs.f16(half %copysign)
+  %fneg.fabs.copysign = fneg fast half %fabs.copysign
+  ret half %fneg.fabs.copysign
+}
+
+define half @fabs_copysign_flags(half %x, half %y) {
+; CHECK-LABEL: @fabs_copysign_flags(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call nnan half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call nnan nsz half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    ret half [[FABS_COPYSIGN]]
+;
+  %copysign = call nnan half @llvm.copysign.f16(half %x, half %y)
+  %fabs.copysign = call nsz nnan half @llvm.fabs.f16(half %copysign)
+  ret half %fabs.copysign
+}
+
+define half @fabs_copysign_all_flags(half %x, half %y) {
+; CHECK-LABEL: @fabs_copysign_all_flags(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call fast half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call fast half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    ret half [[FABS_COPYSIGN]]
+;
+  %copysign = call fast half @llvm.copysign.f16(half %x, half %y)
+  %fabs.copysign = call fast half @llvm.fabs.f16(half %copysign)
+  ret half %fabs.copysign
+}
+
+define half @fabs_copysign_no_flags_copysign_user(half %x, half %y, ptr %ptr) {
+; CHECK-LABEL: @fabs_copysign_no_flags_copysign_user(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    store half [[COPYSIGN]], ptr [[PTR:%.*]], align 2
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call fast half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    ret half [[FABS_COPYSIGN]]
+;
+  %copysign = call half @llvm.copysign.f16(half %x, half %y)
+  store half %copysign, ptr %ptr
+  %fabs.copysign = call fast half @llvm.fabs.f16(half %copysign)
+  ret half %fabs.copysign
+}
+
+define half @fneg_fabs_copysign_drop_flags(half %x, half %y) {
+; CHECK-LABEL: @fneg_fabs_copysign_drop_flags(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call nnan half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call ninf half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    [[FNEG_FABS_COPYSIGN:%.*]] = fneg nsz half [[FABS_COPYSIGN]]
+; CHECK-NEXT:    ret half [[FNEG_FABS_COPYSIGN]]
+;
+  %copysign = call nnan half @llvm.copysign.f16(half %x, half %y)
+  %fabs.copysign = call ninf half @llvm.fabs.f16(half %copysign)
+  %fneg.fabs.copysign = fneg nsz half %fabs.copysign
+  ret half %fneg.fabs.copysign
+}
+
+define half @fneg_copysign_multi_use(half %x, half %y, ptr %ptr) {
+; CHECK-LABEL: @fneg_copysign_multi_use(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    store half [[COPYSIGN]], ptr [[PTR:%.*]], align 2
+; CHECK-NEXT:    [[FNEG_COPYSIGN:%.*]] = fneg half [[COPYSIGN]]
+; CHECK-NEXT:    ret half [[FNEG_COPYSIGN]]
+;
+  %copysign = call half @llvm.copysign.f16(half %x, half %y)
+  store half %copysign, ptr %ptr
+  %fneg.copysign = fneg half %copysign
+  ret half %fneg.copysign
+}
+
+define half @fabs_copysign_multi_use(half %x, half %y, ptr %ptr) {
+; CHECK-LABEL: @fabs_copysign_multi_use(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    store half [[COPYSIGN]], ptr [[PTR:%.*]], align 2
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    ret half [[FABS_COPYSIGN]]
+;
+  %copysign = call half @llvm.copysign.f16(half %x, half %y)
+  store half %copysign, ptr %ptr
+  %fabs.copysign = call half @llvm.fabs.f16(half %copysign)
+  ret half %fabs.copysign
+}
+
+define half @fabs_flags_copysign_multi_use(half %x, half %y, ptr %ptr) {
+; CHECK-LABEL: @fabs_flags_copysign_multi_use(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    store half [[COPYSIGN]], ptr [[PTR:%.*]], align 2
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call nnan ninf half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    ret half [[FABS_COPYSIGN]]
+;
+  %copysign = call half @llvm.copysign.f16(half %x, half %y)
+  store half %copysign, ptr %ptr
+  %fabs.copysign = call ninf nnan half @llvm.fabs.f16(half %copysign)
+  ret half %fabs.copysign
+}
+
+define half @fneg_fabs_copysign_multi_use_fabs(half %x, half %y, ptr %ptr) {
+; CHECK-LABEL: @fneg_fabs_copysign_multi_use_fabs(
+; CHECK-NEXT:    [[COPYSIGN:%.*]] = call half @llvm.copysign.f16(half [[X:%.*]], half [[Y:%.*]])
+; CHECK-NEXT:    [[FABS_COPYSIGN:%.*]] = call half @llvm.fabs.f16(half [[COPYSIGN]])
+; CHECK-NEXT:    store half [[FABS_COPYSIGN]], ptr [[PTR:%.*]], align 2
+; CHECK-NEXT:    ret half [[FABS_COPYSIGN]]
+;
+  %copysign = call half @llvm.copysign.f16(half %x, half %y)
+  %fabs.copysign = call half @llvm.fabs.f16(half %copysign)
+  store half %fabs.copysign, ptr %ptr
+  ret half %fabs.copysign
+}
+
+declare half @llvm.fabs.f16(half)
+declare <2 x half> @llvm.fabs.v2f16(<2 x half>)
+declare half @llvm.copysign.f16(half, half)
+declare <2 x half> @llvm.copysign.v2f16(<2 x half>, <2 x half>)