76 lines · plain
1; RUN: llc < %s -mtriple=nvptx64 | FileCheck %s2; RUN: %if ptxas %{ llc < %s -mtriple=nvptx64 | %ptxas-verify %}3 4; CHECK: .b8 half_array[8] = {1, 2, 3, 4, 5, 6, 7, 8};5@"half_array" = addrspace(1) constant [4 x half]6 [half 0xH0201, half 0xH0403, half 0xH0605, half 0xH0807]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 half, ptr addrspace(1) %in13 store half %val, ptr addrspace(1) %out14 ret void15}16 17define void @test_bitcast_from_half(ptr addrspace(1) %in, ptr addrspace(1) %out) {18; CHECK-LABEL: @test_bitcast_from_half19; CHECK: ld.global.b16 [[TMP:%rs[0-9]+]], [{{%rd[0-9]+}}]20; CHECK: st.global.b16 [{{%rd[0-9]+}}], [[TMP]]21 %val = load half, ptr addrspace(1) %in22 %val_int = bitcast half %val to i1623 store i16 %val_int, ptr addrspace(1) %out24 ret void25}26 27define void @test_bitcast_to_half(ptr addrspace(1) %out, ptr addrspace(1) %in) {28; CHECK-LABEL: @test_bitcast_to_half29; 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 half33 store half %val_fp, ptr addrspace(1) %out34 ret void35}36 37define void @test_extend32(ptr addrspace(1) %in, ptr addrspace(1) %out) {38; CHECK-LABEL: @test_extend3239; CHECK: cvt.f32.f1640 41 %val16 = load half, ptr addrspace(1) %in42 %val32 = fpext half %val16 to float43 store float %val32, ptr addrspace(1) %out44 ret void45}46 47define void @test_extend64(ptr addrspace(1) %in, ptr addrspace(1) %out) {48; CHECK-LABEL: @test_extend6449; CHECK: cvt.f64.f1650 51 %val16 = load half, ptr addrspace(1) %in52 %val64 = fpext half %val16 to double53 store double %val64, ptr addrspace(1) %out54 ret void55}56 57define void @test_trunc32(ptr addrspace(1) %in, ptr addrspace(1) %out) {58; CHECK-LABEL: test_trunc3259; CHECK: cvt.rn.f16.f3260 61 %val32 = load float, ptr addrspace(1) %in62 %val16 = fptrunc float %val32 to half63 store half %val16, ptr addrspace(1) %out64 ret void65}66 67define void @test_trunc64(ptr addrspace(1) %in, ptr addrspace(1) %out) {68; CHECK-LABEL: @test_trunc6469; CHECK: cvt.rn.f16.f6470 71 %val32 = load double, ptr addrspace(1) %in72 %val16 = fptrunc double %val32 to half73 store half %val16, ptr addrspace(1) %out74 ret void75}76