llvm/test/CodeGen/NVPTX/half.ll

   1 ; RUN: llc < %s -march=nvptx64 | FileCheck %s
   2 ; RUN: %if ptxas %{ llc < %s -march=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
   8 define 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 half, ptr addrspace(1) %in
  13   store half %val, ptr addrspace(1) %out
  14   ret void
  15 }
  16
  17 define void @test_bitcast_from_half(ptr addrspace(1) %in, ptr addrspace(1) %out) {
  18 ; CHECK-LABEL: @test_bitcast_from_half
  19 ; 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) %in
  22   %val_int = bitcast half %val to i16
  23   store i16 %val_int, ptr addrspace(1) %out
  24   ret void
  25 }
  26
  27 define void @test_bitcast_to_half(ptr addrspace(1) %out, ptr addrspace(1) %in) {
  28 ; CHECK-LABEL: @test_bitcast_to_half
  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 half
  33   store half %val_fp, ptr addrspace(1) %out
  34   ret void
  35 }
  36
  37 define void @test_extend32(ptr addrspace(1) %in, ptr addrspace(1) %out) {
  38 ; CHECK-LABEL: @test_extend32
  39 ; CHECK: cvt.f32.f16
  40
  41   %val16 = load half, ptr addrspace(1) %in
  42   %val32 = fpext half %val16 to float
  43   store float %val32, ptr addrspace(1) %out
  44   ret void
  45 }
  46
  47 define void @test_extend64(ptr addrspace(1) %in, ptr addrspace(1) %out) {
  48 ; CHECK-LABEL: @test_extend64
  49 ; CHECK: cvt.f64.f16
  50
  51   %val16 = load half, ptr addrspace(1) %in
  52   %val64 = fpext half %val16 to double
  53   store double %val64, ptr addrspace(1) %out
  54   ret void
  55 }
  56
  57 define void @test_trunc32(ptr addrspace(1) %in, ptr addrspace(1) %out) {
  58 ; CHECK-LABEL: test_trunc32
  59 ; CHECK: cvt.rn.f16.f32
  60
  61   %val32 = load float, ptr addrspace(1) %in
  62   %val16 = fptrunc float %val32 to half
  63   store half %val16, ptr addrspace(1) %out
  64   ret void
  65 }
  66
  67 define void @test_trunc64(ptr addrspace(1) %in, ptr addrspace(1) %out) {
  68 ; CHECK-LABEL: @test_trunc64
  69 ; CHECK: cvt.rn.f16.f64
  70
  71   %val32 = load double, ptr addrspace(1) %in
  72   %val16 = fptrunc double %val32 to half
  73   store half %val16, ptr addrspace(1) %out
  74   ret void
  75 }