xref: /llvm-project/llvm/test/CodeGen/NVPTX/fp16.ll (revision b279f6b098d3849f7f1c1f539b108307d5f8ae2d)
1; RUN: llc -mtriple=nvptx64 -verify-machineinstrs < %s | FileCheck %s
2; RUN: %if ptxas %{ llc -mtriple=nvptx64 -verify-machineinstrs < %s | %ptxas-verify %}
3
4declare float @llvm.convert.from.fp16.f32(i16) nounwind readnone
5declare double @llvm.convert.from.fp16.f64(i16) nounwind readnone
6declare i16 @llvm.convert.to.fp16.f32(float) nounwind readnone
7declare i16 @llvm.convert.to.fp16.f64(double) nounwind readnone
8
9; CHECK-LABEL: @test_convert_fp16_to_fp32
10; CHECK: cvt.f32.f16
11define void @test_convert_fp16_to_fp32(ptr addrspace(1) noalias %out, ptr addrspace(1) noalias %in) nounwind {
12  %val = load i16, ptr addrspace(1) %in, align 2
13  %cvt = call float @llvm.convert.from.fp16.f32(i16 %val) nounwind readnone
14  store float %cvt, ptr addrspace(1) %out, align 4
15  ret void
16}
17
18
19; CHECK-LABEL: @test_convert_fp16_to_fp64
20; CHECK: cvt.f64.f16
21define void @test_convert_fp16_to_fp64(ptr addrspace(1) noalias %out, ptr addrspace(1) noalias %in) nounwind {
22  %val = load i16, ptr addrspace(1) %in, align 2
23  %cvt = call double @llvm.convert.from.fp16.f64(i16 %val) nounwind readnone
24  store double %cvt, ptr addrspace(1) %out, align 4
25  ret void
26}
27
28
29; CHECK-LABEL: @test_convert_fp32_to_fp16
30; CHECK: cvt.rn.f16.f32
31define void @test_convert_fp32_to_fp16(ptr addrspace(1) noalias %out, ptr addrspace(1) noalias %in) nounwind {
32  %val = load float, ptr addrspace(1) %in, align 2
33  %cvt = call i16 @llvm.convert.to.fp16.f32(float %val) nounwind readnone
34  store i16 %cvt, ptr addrspace(1) %out, align 4
35  ret void
36}
37
38
39; CHECK-LABEL: @test_convert_fp64_to_fp16
40; CHECK: cvt.rn.f16.f64
41define void @test_convert_fp64_to_fp16(ptr addrspace(1) noalias %out, ptr addrspace(1) noalias %in) nounwind {
42  %val = load double, ptr addrspace(1) %in, align 2
43  %cvt = call i16 @llvm.convert.to.fp16.f64(double %val) nounwind readnone
44  store i16 %cvt, ptr addrspace(1) %out, align 4
45  ret void
46}
47