36 lines · plain
1; RUN: llc < %s -mtriple=nvptx64 | FileCheck %s2; 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_store10; 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) %in13 store bfloat %val, ptr addrspace(1) %out14 ret void15}16 17define void @test_bitcast_from_bfloat(ptr addrspace(1) %in, ptr addrspace(1) %out) {18; CHECK-LABEL: @test_bitcast_from_bfloat19; 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) %in22 %val_int = bitcast bfloat %val to i1623 store i16 %val_int, ptr addrspace(1) %out24 ret void25}26 27define void @test_bitcast_to_bfloat(ptr addrspace(1) %out, ptr addrspace(1) %in) {28; CHECK-LABEL: @test_bitcast_to_bfloat29; CHECK: ld.global.b16 [[TMP:%rs[0-9]+]], [{{%rd[0-9]+}}]30; CHECK: st.global.b16 [{{%rd[0-9]+}}], [[TMP]]31 %val = load i16, ptr addrspace(1) %in32 %val_fp = bitcast i16 %val to bfloat33 store bfloat %val_fp, ptr addrspace(1) %out34 ret void35}36