llvm-project/llvm/test/CodeGen/SystemZ/fp-half-vector-move.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

123 lines
3.8 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 6
; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s
; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z16 | FileCheck %s --check-prefix=VECTOR
;
; Test insertions into fp16 vectors.
define <8 x half> @f0(half %val) {
; CHECK-LABEL: f0:
; CHECK: # %bb.0:
; CHECK-NEXT: # kill: def $f0h killed $f0h def $f0d
; CHECK-NEXT: lgdr %r0, %f0
; CHECK-NEXT: srlg %r0, %r0, 48
; CHECK-NEXT: sth %r0, 4(%r2)
; CHECK-NEXT: br %r14
;
; VECTOR-LABEL: f0:
; VECTOR: # %bb.0:
; VECTOR-NEXT: # kill: def $f0h killed $f0h def $v0
; VECTOR-NEXT: vreph %v24, %v0, 0
; VECTOR-NEXT: br %r14
%ret = insertelement <8 x half> poison, half %val, i32 2
ret <8 x half> %ret
}
define <8 x half> @f1(half %val) {
; CHECK-LABEL: f1:
; CHECK: # %bb.0:
; CHECK-NEXT: # kill: def $f0h killed $f0h def $f0d
; CHECK-NEXT: lgdr %r0, %f0
; CHECK-NEXT: srlg %r0, %r0, 48
; CHECK-NEXT: sth %r0, 6(%r2)
; CHECK-NEXT: sth %r0, 4(%r2)
; CHECK-NEXT: br %r14
;
; VECTOR-LABEL: f1:
; VECTOR: # %bb.0:
; VECTOR-NEXT: # kill: def $f0h killed $f0h def $v0
; VECTOR-NEXT: vreph %v24, %v0, 0
; VECTOR-NEXT: br %r14
%v0 = insertelement <8 x half> poison, half %val, i32 2
%ret = insertelement <8 x half> %v0, half %val, i32 3
ret <8 x half> %ret
}
define <8 x half> @f2(half %val0, half %val1) {
; CHECK-LABEL: f2:
; CHECK: # %bb.0:
; CHECK-NEXT: # kill: def $f2h killed $f2h def $f2d
; CHECK-NEXT: # kill: def $f0h killed $f0h def $f0d
; CHECK-NEXT: lgdr %r0, %f2
; CHECK-NEXT: srlg %r0, %r0, 48
; CHECK-NEXT: sth %r0, 6(%r2)
; CHECK-NEXT: lgdr %r0, %f0
; CHECK-NEXT: srlg %r0, %r0, 48
; CHECK-NEXT: sth %r0, 4(%r2)
; CHECK-NEXT: br %r14
;
; VECTOR-LABEL: f2:
; VECTOR: # %bb.0:
; VECTOR-NEXT: # kill: def $f2h killed $f2h def $v2
; VECTOR-NEXT: # kill: def $f0h killed $f0h def $v0
; VECTOR-NEXT: vmrhh %v0, %v0, %v2
; VECTOR-NEXT: vmrhf %v0, %v0, %v0
; VECTOR-NEXT: vmrhg %v24, %v0, %v0
; VECTOR-NEXT: br %r14
%v0 = insertelement <8 x half> poison, half %val0, i32 2
%ret = insertelement <8 x half> %v0, half %val1, i32 3
ret <8 x half> %ret
}
define <8 x half> @f3(half %val0, half %val1) {
; CHECK-LABEL: f3:
; CHECK: # %bb.0:
; CHECK-NEXT: # kill: def $f2h killed $f2h def $f2d
; CHECK-NEXT: # kill: def $f0h killed $f0h def $f0d
; CHECK-NEXT: lgdr %r0, %f2
; CHECK-NEXT: srlg %r0, %r0, 48
; CHECK-NEXT: sth %r0, 10(%r2)
; CHECK-NEXT: lgdr %r1, %f0
; CHECK-NEXT: srlg %r1, %r1, 48
; CHECK-NEXT: sth %r1, 8(%r2)
; CHECK-NEXT: sth %r0, 6(%r2)
; CHECK-NEXT: sth %r1, 4(%r2)
; CHECK-NEXT: br %r14
;
; VECTOR-LABEL: f3:
; VECTOR: # %bb.0:
; VECTOR-NEXT: # kill: def $f2h killed $f2h def $v2
; VECTOR-NEXT: # kill: def $f0h killed $f0h def $v0
; VECTOR-NEXT: vmrhh %v0, %v0, %v2
; VECTOR-NEXT: vmrhf %v0, %v0, %v0
; VECTOR-NEXT: vmrhg %v24, %v0, %v0
; VECTOR-NEXT: br %r14
%v0 = insertelement <8 x half> poison, half %val0, i32 2
%v1 = insertelement <8 x half> %v0, half %val1, i32 3
%v2 = insertelement <8 x half> %v1, half %val0, i32 4
%ret = insertelement <8 x half> %v2, half %val1, i32 5
ret <8 x half> %ret
}
; Test creation of vregs where the arg gets one VR128 which is split into two
; VR16 parts.
define <2 x half> @f4(<2 x half> %0) {
; CHECK-LABEL: f4:
; CHECK: # %bb.0: # %entry
; CHECK-NEXT: lzer %f0
; CHECK-NEXT: br %r14
;
; VECTOR-LABEL: f4:
; VECTOR: # %bb.0: # %entry
; VECTOR-NEXT: vreph %v0, %v24, 1
; VECTOR-NEXT: vuplhh %v0, %v0
; VECTOR-NEXT: vmrhf %v0, %v0, %v0
; VECTOR-NEXT: vmrhg %v24, %v0, %v0
; VECTOR-NEXT: br %r14
entry:
br label %body
body: ; preds = %entry
%2 = insertelement <2 x half> %0, half 0x0000, i64 0
ret <2 x half> %2
}