Files
llvm-project/llvm/test/CodeGen/SystemZ/vec-cmp-09.ll
Jonas Paulsson c999e9a4fe [SystemZ] Support fp16 vector ABI and basic codegen. (#171066)
- Make v8f16 a legal type so that arguments can be passed in vector
registers. Handle fp16 vectors so that they have the same ABI as other
fp vectors.

- Set the preferred vector action for fp16 vectors to "split". This will
scalarize all operations, which is not always necessary (like with
memory operations), but it avoids the superfluous operations that result
after first widening and then scalarizing a narrow vector (like v4f16).

Fixes #168992
2026-01-26 13:42:25 -06:00

288 lines
10 KiB
LLVM

; Test usage of VBLEND on z17.
;
; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z17 | FileCheck %s
define <16 x i8> @f1(<16 x i8> %val1, <16 x i8> %val2, <16 x i8> %val3) {
; CHECK-LABEL: f1:
; CHECK: vblendb %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%cmp = icmp slt <16 x i8> %val1, zeroinitializer
%ret = select <16 x i1> %cmp, <16 x i8> %val2, <16 x i8> %val3
ret <16 x i8> %ret
}
define <16 x i8> @f2(<16 x i8> %val1, <16 x i8> %val2, <16 x i8> %val3) {
; CHECK-LABEL: f2:
; CHECK: vblendb %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%cmp = icmp sge <16 x i8> %val1, zeroinitializer
%ret = select <16 x i1> %cmp, <16 x i8> %val2, <16 x i8> %val3
ret <16 x i8> %ret
}
define <16 x i8> @f3(<16 x i8> %val1, <16 x i8> %val2, <16 x i8> %val3) {
; CHECK-LABEL: f3:
; CHECK: vblendb %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%mask = and <16 x i8> %val1, <i8 128, i8 128, i8 128, i8 128,
i8 128, i8 128, i8 128, i8 128,
i8 128, i8 128, i8 128, i8 128,
i8 128, i8 128, i8 128, i8 128>;
%cmp = icmp ne <16 x i8> %mask, zeroinitializer
%ret = select <16 x i1> %cmp, <16 x i8> %val2, <16 x i8> %val3
ret <16 x i8> %ret
}
define <16 x i8> @f4(<16 x i8> %val1, <16 x i8> %val2, <16 x i8> %val3) {
; CHECK-LABEL: f4:
; CHECK: vblendb %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%mask = and <16 x i8> %val1, <i8 128, i8 128, i8 128, i8 128,
i8 128, i8 128, i8 128, i8 128,
i8 128, i8 128, i8 128, i8 128,
i8 128, i8 128, i8 128, i8 128>;
%cmp = icmp eq <16 x i8> %mask, zeroinitializer
%ret = select <16 x i1> %cmp, <16 x i8> %val2, <16 x i8> %val3
ret <16 x i8> %ret
}
define <8 x i16> @f5(<8 x i16> %val1, <8 x i16> %val2, <8 x i16> %val3) {
; CHECK-LABEL: f5:
; CHECK: vblendh %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%cmp = icmp slt <8 x i16> %val1, zeroinitializer
%ret = select <8 x i1> %cmp, <8 x i16> %val2, <8 x i16> %val3
ret <8 x i16> %ret
}
define <8 x i16> @f6(<8 x i16> %val1, <8 x i16> %val2, <8 x i16> %val3) {
; CHECK-LABEL: f6:
; CHECK: vblendh %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%cmp = icmp sge <8 x i16> %val1, zeroinitializer
%ret = select <8 x i1> %cmp, <8 x i16> %val2, <8 x i16> %val3
ret <8 x i16> %ret
}
define <8 x i16> @f7(<8 x i16> %val1, <8 x i16> %val2, <8 x i16> %val3) {
; CHECK-LABEL: f7:
; CHECK: vblendh %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%mask = and <8 x i16> %val1, <i16 32768, i16 32768, i16 32768, i16 32768,
i16 32768, i16 32768, i16 32768, i16 32768>;
%cmp = icmp ne <8 x i16> %mask, zeroinitializer
%ret = select <8 x i1> %cmp, <8 x i16> %val2, <8 x i16> %val3
ret <8 x i16> %ret
}
define <8 x i16> @f8(<8 x i16> %val1, <8 x i16> %val2, <8 x i16> %val3) {
; CHECK-LABEL: f8:
; CHECK: vblendh %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%mask = and <8 x i16> %val1, <i16 32768, i16 32768, i16 32768, i16 32768,
i16 32768, i16 32768, i16 32768, i16 32768>;
%cmp = icmp eq <8 x i16> %mask, zeroinitializer
%ret = select <8 x i1> %cmp, <8 x i16> %val2, <8 x i16> %val3
ret <8 x i16> %ret
}
define <4 x i32> @f9(<4 x i32> %val1, <4 x i32> %val2, <4 x i32> %val3) {
; CHECK-LABEL: f9:
; CHECK: vblendf %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%cmp = icmp slt <4 x i32> %val1, zeroinitializer
%ret = select <4 x i1> %cmp, <4 x i32> %val2, <4 x i32> %val3
ret <4 x i32> %ret
}
define <4 x i32> @f10(<4 x i32> %val1, <4 x i32> %val2, <4 x i32> %val3) {
; CHECK-LABEL: f10:
; CHECK: vblendf %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%cmp = icmp sge <4 x i32> %val1, zeroinitializer
%ret = select <4 x i1> %cmp, <4 x i32> %val2, <4 x i32> %val3
ret <4 x i32> %ret
}
define <4 x i32> @f11(<4 x i32> %val1, <4 x i32> %val2, <4 x i32> %val3) {
; CHECK-LABEL: f11:
; CHECK: vblendf %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%mask = and <4 x i32> %val1, <i32 2147483648, i32 2147483648,
i32 2147483648, i32 2147483648>;
%cmp = icmp ne <4 x i32> %mask, zeroinitializer
%ret = select <4 x i1> %cmp, <4 x i32> %val2, <4 x i32> %val3
ret <4 x i32> %ret
}
define <4 x i32> @f12(<4 x i32> %val1, <4 x i32> %val2, <4 x i32> %val3) {
; CHECK-LABEL: f12:
; CHECK: vblendf %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%mask = and <4 x i32> %val1, <i32 2147483648, i32 2147483648,
i32 2147483648, i32 2147483648>;
%cmp = icmp eq <4 x i32> %mask, zeroinitializer
%ret = select <4 x i1> %cmp, <4 x i32> %val2, <4 x i32> %val3
ret <4 x i32> %ret
}
define <2 x i64> @f13(<2 x i64> %val1, <2 x i64> %val2, <2 x i64> %val3) {
; CHECK-LABEL: f13:
; CHECK: vblendg %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%cmp = icmp slt <2 x i64> %val1, zeroinitializer
%ret = select <2 x i1> %cmp, <2 x i64> %val2, <2 x i64> %val3
ret <2 x i64> %ret
}
define <2 x i64> @f14(<2 x i64> %val1, <2 x i64> %val2, <2 x i64> %val3) {
; CHECK-LABEL: f14:
; CHECK: vblendg %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%cmp = icmp sge <2 x i64> %val1, zeroinitializer
%ret = select <2 x i1> %cmp, <2 x i64> %val2, <2 x i64> %val3
ret <2 x i64> %ret
}
define <2 x i64> @f15(<2 x i64> %val1, <2 x i64> %val2, <2 x i64> %val3) {
; CHECK-LABEL: f15:
; CHECK: vblendg %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%mask = and <2 x i64> %val1, <i64 9223372036854775808,
i64 9223372036854775808>;
%cmp = icmp ne <2 x i64> %mask, zeroinitializer
%ret = select <2 x i1> %cmp, <2 x i64> %val2, <2 x i64> %val3
ret <2 x i64> %ret
}
define <2 x i64> @f16(<2 x i64> %val1, <2 x i64> %val2, <2 x i64> %val3) {
; CHECK-LABEL: f16:
; CHECK: vblendg %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%mask = and <2 x i64> %val1, <i64 9223372036854775808,
i64 9223372036854775808>;
%cmp = icmp eq <2 x i64> %mask, zeroinitializer
%ret = select <2 x i1> %cmp, <2 x i64> %val2, <2 x i64> %val3
ret <2 x i64> %ret
}
define <8 x half> @f17(<8 x i16> %val1, <8 x half> %val2, <8 x half> %val3) {
; CHECK-LABEL: f17:
; CHECK: vblendh %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%cmp = icmp slt <8 x i16> %val1, zeroinitializer
%ret = select <8 x i1> %cmp, <8 x half> %val2, <8 x half> %val3
ret <8 x half> %ret
}
define <8 x half> @f18(<8 x i16> %val1, <8 x half> %val2, <8 x half> %val3) {
; CHECK-LABEL: f18:
; CHECK: vblendh %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%cmp = icmp sge <8 x i16> %val1, zeroinitializer
%ret = select <8 x i1> %cmp, <8 x half> %val2, <8 x half> %val3
ret <8 x half> %ret
}
define <8 x half> @f19(<8 x i16> %val1, <8 x half> %val2, <8 x half> %val3) {
; CHECK-LABEL: f19:
; CHECK: vblendh %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%mask = and <8 x i16> %val1, <i16 32768, i16 32768, i16 32768, i16 32768,
i16 32768, i16 32768, i16 32768, i16 32768>
%cmp = icmp ne <8 x i16> %mask, zeroinitializer
%ret = select <8 x i1> %cmp, <8 x half> %val2, <8 x half> %val3
ret <8 x half> %ret
}
define <8 x half> @f20(<8 x i16> %val1, <8 x half> %val2, <8 x half> %val3) {
; CHECK-LABEL: f20:
; CHECK: vblendh %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%mask = and <8 x i16> %val1, <i16 32768, i16 32768, i16 32768, i16 32768,
i16 32768, i16 32768, i16 32768, i16 32768>
%cmp = icmp eq <8 x i16> %mask, zeroinitializer
%ret = select <8 x i1> %cmp, <8 x half> %val2, <8 x half> %val3
ret <8 x half> %ret
}
define <4 x float> @f21(<4 x i32> %val1, <4 x float> %val2, <4 x float> %val3) {
; CHECK-LABEL: f21:
; CHECK: vblendf %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%cmp = icmp slt <4 x i32> %val1, zeroinitializer
%ret = select <4 x i1> %cmp, <4 x float> %val2, <4 x float> %val3
ret <4 x float> %ret
}
define <4 x float> @f22(<4 x i32> %val1, <4 x float> %val2, <4 x float> %val3) {
; CHECK-LABEL: f22:
; CHECK: vblendf %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%cmp = icmp sge <4 x i32> %val1, zeroinitializer
%ret = select <4 x i1> %cmp, <4 x float> %val2, <4 x float> %val3
ret <4 x float> %ret
}
define <4 x float> @f23(<4 x i32> %val1, <4 x float> %val2, <4 x float> %val3) {
; CHECK-LABEL: f23:
; CHECK: vblendf %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%mask = and <4 x i32> %val1, <i32 2147483648, i32 2147483648,
i32 2147483648, i32 2147483648>;
%cmp = icmp ne <4 x i32> %mask, zeroinitializer
%ret = select <4 x i1> %cmp, <4 x float> %val2, <4 x float> %val3
ret <4 x float> %ret
}
define <4 x float> @f24(<4 x i32> %val1, <4 x float> %val2, <4 x float> %val3) {
; CHECK-LABEL: f24:
; CHECK: vblendf %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%mask = and <4 x i32> %val1, <i32 2147483648, i32 2147483648,
i32 2147483648, i32 2147483648>;
%cmp = icmp eq <4 x i32> %mask, zeroinitializer
%ret = select <4 x i1> %cmp, <4 x float> %val2, <4 x float> %val3
ret <4 x float> %ret
}
define <2 x double> @f25(<2 x i64> %val1, <2 x double> %val2, <2 x double> %val3) {
; CHECK-LABEL: f25:
; CHECK: vblendg %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%cmp = icmp slt <2 x i64> %val1, zeroinitializer
%ret = select <2 x i1> %cmp, <2 x double> %val2, <2 x double> %val3
ret <2 x double> %ret
}
define <2 x double> @f26(<2 x i64> %val1, <2 x double> %val2, <2 x double> %val3) {
; CHECK-LABEL: f26:
; CHECK: vblendg %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%cmp = icmp sge <2 x i64> %val1, zeroinitializer
%ret = select <2 x i1> %cmp, <2 x double> %val2, <2 x double> %val3
ret <2 x double> %ret
}
define <2 x double> @f27(<2 x i64> %val1, <2 x double> %val2, <2 x double> %val3) {
; CHECK-LABEL: f27:
; CHECK: vblendg %v24, %v26, %v28, %v24
; CHECK-NEXT: br %r14
%mask = and <2 x i64> %val1, <i64 9223372036854775808,
i64 9223372036854775808>;
%cmp = icmp ne <2 x i64> %mask, zeroinitializer
%ret = select <2 x i1> %cmp, <2 x double> %val2, <2 x double> %val3
ret <2 x double> %ret
}
define <2 x double> @f28(<2 x i64> %val1, <2 x double> %val2, <2 x double> %val3) {
; CHECK-LABEL: f28:
; CHECK: vblendg %v24, %v28, %v26, %v24
; CHECK-NEXT: br %r14
%mask = and <2 x i64> %val1, <i64 9223372036854775808,
i64 9223372036854775808>;
%cmp = icmp eq <2 x i64> %mask, zeroinitializer
%ret = select <2 x i1> %cmp, <2 x double> %val2, <2 x double> %val3
ret <2 x double> %ret
}