; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py ; RUN: llc -mattr=+sve2,+bf16 < %s -o - | FileCheck %s target triple = "aarch64-unknown-linux-gnu" ; ; llvm.copysign.nxv2f16 ; define @copysign_nxv2f16_nxv2f16( %a, %b) { ; CHECK-LABEL: copysign_nxv2f16_nxv2f16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %r = call @llvm.copysign.nxv2f16( %a, %b) ret %r } define @copysign_nxv2f16_nxv2f32( %a, %b) { ; CHECK-LABEL: copysign_nxv2f16_nxv2f32: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: fcvt z1.h, p0/m, z1.s ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv2f16( %a, %b.trunc) ret %r } define @copysign_nxv2f16_nxv2f64( %a, %b) { ; CHECK-LABEL: copysign_nxv2f16_nxv2f64: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: fcvt z1.h, p0/m, z1.d ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv2f16( %a, %b.trunc) ret %r } ; ; llvm.copysign.nxv4f16 ; define @copysign_nxv4f16_nxv4f16( %a, %b) { ; CHECK-LABEL: copysign_nxv4f16_nxv4f16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %r = call @llvm.copysign.nxv4f16( %a, %b) ret %r } define @copysign_nxv4f16_nxv4f32( %a, %b) { ; CHECK-LABEL: copysign_nxv4f16_nxv4f32: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.s ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: fcvt z1.h, p0/m, z1.s ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv4f16( %a, %b.trunc) ret %r } define @copysign_nxv4f16_nxv4f64( %a, %b) { ; CHECK-LABEL: copysign_nxv4f16_nxv4f64: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z3.h, #32767 // =0x7fff ; CHECK-NEXT: fcvt z2.h, p0/m, z2.d ; CHECK-NEXT: fcvt z1.h, p0/m, z1.d ; CHECK-NEXT: uzp1 z1.s, z1.s, z2.s ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z3.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv4f16( %a, %b.trunc) ret %r } ; ; llvm.copysign.nxv8f16 ; define @copysign_nxv8f16_nxv8f16( %a, %b) { ; CHECK-LABEL: copysign_nxv8f16_nxv8f16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %r = call @llvm.copysign.nxv8f16( %a, %b) ret %r } define @copysign_nxv8f16_nxv8f32( %a, %b) { ; CHECK-LABEL: copysign_nxv8f16_nxv8f32: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.s ; CHECK-NEXT: mov z3.h, #32767 // =0x7fff ; CHECK-NEXT: fcvt z2.h, p0/m, z2.s ; CHECK-NEXT: fcvt z1.h, p0/m, z1.s ; CHECK-NEXT: uzp1 z1.h, z1.h, z2.h ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z3.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv8f16( %a, %b.trunc) ret %r } define @copysign_nxv8f16_nxv8f64( %a, %b) { ; CHECK-LABEL: copysign_nxv8f16_nxv8f64: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: fcvt z4.h, p0/m, z4.d ; CHECK-NEXT: fcvt z3.h, p0/m, z3.d ; CHECK-NEXT: fcvt z2.h, p0/m, z2.d ; CHECK-NEXT: fcvt z1.h, p0/m, z1.d ; CHECK-NEXT: uzp1 z3.s, z3.s, z4.s ; CHECK-NEXT: uzp1 z1.s, z1.s, z2.s ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: uzp1 z1.h, z1.h, z3.h ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv8f16( %a, %b.trunc) ret %r } ; ; llvm.copysign.nxv2f32 ; define @copysign_nxv2f32_nxv2f16( %a, %b) { ; CHECK-LABEL: copysign_nxv2f32_nxv2f16: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z2.s, #0x7fffffff ; CHECK-NEXT: fcvt z1.s, p0/m, z1.h ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.ext = fpext %b to %r = call @llvm.copysign.nxv2f32( %a, %b.ext) ret %r } define @copysign_nxv2f32_nxv2f32( %a, %b) { ; CHECK-LABEL: copysign_nxv2f32_nxv2f32: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.s, #0x7fffffff ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %r = call @llvm.copysign.nxv2f32( %a, %b) ret %r } define @copysign_nxv2f32_nxv2f64( %a, %b) { ; CHECK-LABEL: copysign_nxv2f32_nxv2f64: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z2.s, #0x7fffffff ; CHECK-NEXT: fcvt z1.s, p0/m, z1.d ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv2f32( %a, %b.trunc) ret %r } define @copysign_nxv2f32_nxv2bf16( %a, %b) { ; CHECK-LABEL: copysign_nxv2f32_nxv2bf16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.s, #0x7fffffff ; CHECK-NEXT: lsl z1.s, z1.s, #16 ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %tmp0 = fpext %b to %r = call @llvm.copysign.nxv2f32( %a, %tmp0) ret %r } ; ; llvm.copysign.nxv4f32 ; define @copysign_nxv4f32_nxv4f16( %a, %b) { ; CHECK-LABEL: copysign_nxv4f32_nxv4f16: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.s ; CHECK-NEXT: mov z2.s, #0x7fffffff ; CHECK-NEXT: fcvt z1.s, p0/m, z1.h ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.ext = fpext %b to %r = call @llvm.copysign.nxv4f32( %a, %b.ext) ret %r } define @copysign_nxv4f32_nxv4f32( %a, %b) { ; CHECK-LABEL: copysign_nxv4f32_nxv4f32: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.s, #0x7fffffff ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %r = call @llvm.copysign.nxv4f32( %a, %b) ret %r } define @copysign_nxv4f32_nxv4f64( %a, %b) { ; CHECK-LABEL: copysign_nxv4f32_nxv4f64: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z3.s, #0x7fffffff ; CHECK-NEXT: fcvt z2.s, p0/m, z2.d ; CHECK-NEXT: fcvt z1.s, p0/m, z1.d ; CHECK-NEXT: uzp1 z1.s, z1.s, z2.s ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z3.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv4f32( %a, %b.trunc) ret %r } define @copysign_nxv4f32_nxv4bf16( %a, %b) { ; CHECK-LABEL: copysign_nxv4f32_nxv4bf16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.s, #0x7fffffff ; CHECK-NEXT: lsl z1.s, z1.s, #16 ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %tmp0 = fpext %b to %r = call @llvm.copysign.nxv4f32( %a, %tmp0) ret %r } ; ; llvm.copysign.nxv2f64 ; define @copysign_nxv2f64_nxv2f16( %a, %b) { ; CHECK-LABEL: copysign_nxv2f64_nxv2f16: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z2.d, #0x7fffffffffffffff ; CHECK-NEXT: fcvt z1.d, p0/m, z1.h ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.ext = fpext %b to %r = call @llvm.copysign.nxv2f64( %a, %b.ext) ret %r } define @copysign_nxv2f64_nxv2f32( %a, %b) { ; CHECK-LABEL: copysign_nxv2f64_nxv2f32: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z2.d, #0x7fffffffffffffff ; CHECK-NEXT: fcvt z1.d, p0/m, z1.s ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.ext = fpext %b to %r = call @llvm.copysign.nxv2f64( %a, %b.ext) ret %r } define @copysign_nxv2f64_nxv2f64( %a, %b) { ; CHECK-LABEL: copysign_nxv2f64_nxv2f64: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.d, #0x7fffffffffffffff ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %r = call @llvm.copysign.nxv2f64( %a, %b) ret %r } define @copysign_nxv2f64_nxv2bf16( %a, %b) { ; CHECK-LABEL: copysign_nxv2f64_nxv2bf16: ; CHECK: // %bb.0: ; CHECK-NEXT: lsl z1.s, z1.s, #16 ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z2.d, #0x7fffffffffffffff ; CHECK-NEXT: fcvt z1.d, p0/m, z1.s ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.ext = fpext %b to %r = call @llvm.copysign.nxv2f64( %a, %b.ext) ret %r } ; ; llvm.copysign.nxv2bf16 ; define @copysign_nxv2bf16_nxv2bf16( %a, %b) { ; CHECK-LABEL: copysign_nxv2bf16_nxv2bf16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %r = call @llvm.copysign.nxv2bf16( %a, %b) ret %r } define @copysign_nxv2bf16_nxv2f32( %a, %b) { ; CHECK-LABEL: copysign_nxv2bf16_nxv2f32: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: bfcvt z1.h, p0/m, z1.s ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %tmp0 = fptrunc %b to %r = call @llvm.copysign.nxv2bf16( %a, %tmp0) ret %r } define @copysign_nxv2bf16_nxv2f64( %a, %b) { ; CHECK-LABEL: copysign_nxv2bf16_nxv2f64: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: fcvtx z1.s, p0/m, z1.d ; CHECK-NEXT: bfcvt z1.h, p0/m, z1.s ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %tmp0 = fptrunc %b to %r = call @llvm.copysign.nxv2bf16( %a, %tmp0) ret %r } ; ; llvm.copysign.nxv2bf16 ; define @copysign_nxv4bf16_nxv4bf16( %a, %b) { ; CHECK-LABEL: copysign_nxv4bf16_nxv4bf16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %r = call @llvm.copysign.nxv4bf16( %a, %b) ret %r } define @copysign_nxv4bf16_nxv4f32( %a, %b) { ; CHECK-LABEL: copysign_nxv4bf16_nxv4f32: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.s ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: bfcvt z1.h, p0/m, z1.s ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv4bf16( %a, %b.trunc) ret %r } define @copysign_nxv4bf16_nxv4f64( %a, %b) { ; CHECK-LABEL: copysign_nxv4bf16_nxv4f64: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z3.h, #32767 // =0x7fff ; CHECK-NEXT: fcvtx z2.s, p0/m, z2.d ; CHECK-NEXT: fcvtx z1.s, p0/m, z1.d ; CHECK-NEXT: bfcvt z2.h, p0/m, z2.s ; CHECK-NEXT: bfcvt z1.h, p0/m, z1.s ; CHECK-NEXT: uzp1 z1.s, z1.s, z2.s ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z3.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv4bf16( %a, %b.trunc) ret %r } ; ; llvm.copysign.nxv8bf16 ; define @copysign_nxv8bf16_nxv8bf16( %a, %b) { ; CHECK-LABEL: copysign_nxv8bf16_nxv8bf16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %r = call @llvm.copysign.nxv8bf16( %a, %b) ret %r } define @copysign_nxv8bf16_nxv8f32( %a, %b) { ; CHECK-LABEL: copysign_nxv8bf16_nxv8f32: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.s ; CHECK-NEXT: mov z3.h, #32767 // =0x7fff ; CHECK-NEXT: bfcvt z2.h, p0/m, z2.s ; CHECK-NEXT: bfcvt z1.h, p0/m, z1.s ; CHECK-NEXT: uzp1 z1.h, z1.h, z2.h ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z3.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv8bf16( %a, %b.trunc) ret %r } define @copysign_nxv8bf16_nxv8f64( %a, %b) { ; CHECK-LABEL: copysign_nxv8bf16_nxv8f64: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: fcvtx z4.s, p0/m, z4.d ; CHECK-NEXT: fcvtx z3.s, p0/m, z3.d ; CHECK-NEXT: fcvtx z2.s, p0/m, z2.d ; CHECK-NEXT: fcvtx z1.s, p0/m, z1.d ; CHECK-NEXT: bfcvt z4.h, p0/m, z4.s ; CHECK-NEXT: bfcvt z3.h, p0/m, z3.s ; CHECK-NEXT: bfcvt z2.h, p0/m, z2.s ; CHECK-NEXT: bfcvt z1.h, p0/m, z1.s ; CHECK-NEXT: uzp1 z3.s, z3.s, z4.s ; CHECK-NEXT: uzp1 z1.s, z1.s, z2.s ; CHECK-NEXT: mov z2.h, #32767 // =0x7fff ; CHECK-NEXT: uzp1 z1.h, z1.h, z3.h ; CHECK-NEXT: bsl z0.d, z0.d, z1.d, z2.d ; CHECK-NEXT: ret %b.trunc = fptrunc %b to %r = call @llvm.copysign.nxv8bf16( %a, %b.trunc) ret %r } declare @llvm.copysign.nxv2f16( %a, %b) declare @llvm.copysign.nxv4f16( %a, %b) declare @llvm.copysign.nxv8f16( %a, %b) declare @llvm.copysign.nxv2f32( %a, %b) declare @llvm.copysign.nxv4f32( %a, %b) declare @llvm.copysign.nxv2f64( %a, %b) declare @llvm.copysign.nxv2bf16( %a, %b) declare @llvm.copysign.nxv4bf16( %a, %b) declare @llvm.copysign.nxv8bf16( %a, %b)