xref: /llvm-project/llvm/test/CodeGen/NVPTX/bf16.ll (revision b279f6b098d3849f7f1c1f539b108307d5f8ae2d)
1; RUN: llc < %s -mtriple=nvptx64 | FileCheck %s
2; RUN: %if ptxas %{ llc < %s -mtriple=nvptx64 | %ptxas-verify %}
3
4; LDST: .b8 bfloat_array[8] = {1, 2, 3, 4, 5, 6, 7, 8};
5@"bfloat_array" = addrspace(1) constant [4 x bfloat]
6                [bfloat 0xR0201, bfloat 0xR0403, bfloat 0xR0605, bfloat 0xR0807]
7
8define void @test_load_store(ptr addrspace(1) %in, ptr addrspace(1) %out) {
9; CHECK-LABEL: @test_load_store
10; CHECK: ld.global.b16 [[TMP:%rs[0-9]+]], [{{%rd[0-9]+}}]
11; CHECK: st.global.b16 [{{%rd[0-9]+}}], [[TMP]]
12  %val = load bfloat, ptr addrspace(1) %in
13  store bfloat %val, ptr addrspace(1) %out
14  ret void
15}
16
17define void @test_bitcast_from_bfloat(ptr addrspace(1) %in, ptr addrspace(1) %out) {
18; CHECK-LABEL: @test_bitcast_from_bfloat
19; CHECK: ld.global.b16 [[TMP:%rs[0-9]+]], [{{%rd[0-9]+}}]
20; CHECK: st.global.b16 [{{%rd[0-9]+}}], [[TMP]]
21  %val = load bfloat, ptr addrspace(1) %in
22  %val_int = bitcast bfloat %val to i16
23  store i16 %val_int, ptr addrspace(1) %out
24  ret void
25}
26
27define void @test_bitcast_to_bfloat(ptr addrspace(1) %out, ptr addrspace(1) %in) {
28; CHECK-LABEL: @test_bitcast_to_bfloat
29; CHECK: ld.global.u16 [[TMP:%rs[0-9]+]], [{{%rd[0-9]+}}]
30; CHECK: st.global.u16 [{{%rd[0-9]+}}], [[TMP]]
31  %val = load i16, ptr addrspace(1) %in
32  %val_fp = bitcast i16 %val to bfloat
33  store bfloat %val_fp, ptr addrspace(1) %out
34  ret void
35}
36