mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-23 19:23:23 +01:00
f89a861f79
Only scalar half-precision operations are supported at the moment. - Adds general support for 'half' type in NVPTX. - fp16 math operations are supported on sm_53+ GPUs only (can be disabled with --nvptx-no-f16-math). - Type conversions to/from fp16 are supported on all GPU variants. - On GPU variants that do not have full fp16 support (or if it's disabled), fp16 operations are promoted to fp32 and results are converted back to fp16 for storage. Differential Revision: https://reviews.llvm.org/D28540 llvm-svn: 291956
71 lines
2.2 KiB
LLVM
71 lines
2.2 KiB
LLVM
; RUN: llc < %s -march=nvptx | FileCheck %s
|
|
|
|
define void @test_load_store(half addrspace(1)* %in, half addrspace(1)* %out) {
|
|
; CHECK-LABEL: @test_load_store
|
|
; CHECK: ld.global.b16 [[TMP:%h[0-9]+]], [{{%r[0-9]+}}]
|
|
; CHECK: st.global.b16 [{{%r[0-9]+}}], [[TMP]]
|
|
%val = load half, half addrspace(1)* %in
|
|
store half %val, half addrspace(1) * %out
|
|
ret void
|
|
}
|
|
|
|
define void @test_bitcast_from_half(half addrspace(1)* %in, i16 addrspace(1)* %out) {
|
|
; CHECK-LABEL: @test_bitcast_from_half
|
|
; CHECK: ld.global.b16 [[TMP:%h[0-9]+]], [{{%r[0-9]+}}]
|
|
; CHECK: st.global.b16 [{{%r[0-9]+}}], [[TMP]]
|
|
%val = load half, half addrspace(1) * %in
|
|
%val_int = bitcast half %val to i16
|
|
store i16 %val_int, i16 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
define void @test_bitcast_to_half(half addrspace(1)* %out, i16 addrspace(1)* %in) {
|
|
; CHECK-LABEL: @test_bitcast_to_half
|
|
; CHECK: ld.global.u16 [[TMP:%rs[0-9]+]], [{{%r[0-9]+}}]
|
|
; CHECK: st.global.u16 [{{%r[0-9]+}}], [[TMP]]
|
|
%val = load i16, i16 addrspace(1)* %in
|
|
%val_fp = bitcast i16 %val to half
|
|
store half %val_fp, half addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
define void @test_extend32(half addrspace(1)* %in, float addrspace(1)* %out) {
|
|
; CHECK-LABEL: @test_extend32
|
|
; CHECK: cvt.f32.f16
|
|
|
|
%val16 = load half, half addrspace(1)* %in
|
|
%val32 = fpext half %val16 to float
|
|
store float %val32, float addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
define void @test_extend64(half addrspace(1)* %in, double addrspace(1)* %out) {
|
|
; CHECK-LABEL: @test_extend64
|
|
; CHECK: cvt.f64.f16
|
|
|
|
%val16 = load half, half addrspace(1)* %in
|
|
%val64 = fpext half %val16 to double
|
|
store double %val64, double addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
define void @test_trunc32(float addrspace(1)* %in, half addrspace(1)* %out) {
|
|
; CHECK-LABEL: test_trunc32
|
|
; CHECK: cvt.rn.f16.f32
|
|
|
|
%val32 = load float, float addrspace(1)* %in
|
|
%val16 = fptrunc float %val32 to half
|
|
store half %val16, half addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
define void @test_trunc64(double addrspace(1)* %in, half addrspace(1)* %out) {
|
|
; CHECK-LABEL: @test_trunc64
|
|
; CHECK: cvt.rn.f16.f64
|
|
|
|
%val32 = load double, double addrspace(1)* %in
|
|
%val16 = fptrunc double %val32 to half
|
|
store half %val16, half addrspace(1)* %out
|
|
ret void
|
|
}
|