test/CodeGen/NVPTX/half.ll

   1 ; RUN: llc < %s -march=nvptx | FileCheck %s
   2
   3 ; CHECK: .b8 half_array[8] = {1, 2, 3, 4, 5, 6, 7, 8};
   4 @"half_array" = addrspace(1) constant [4 x half]
   5                 [half 0xH0201, half 0xH0403, half 0xH0605, half 0xH0807]
   6
   7 define void @test_load_store(half addrspace(1)* %in, half addrspace(1)* %out) {
   8 ; CHECK-LABEL: @test_load_store
   9 ; CHECK: ld.global.b16 [[TMP:%h[0-9]+]], [{{%r[0-9]+}}]
  10 ; CHECK: st.global.b16 [{{%r[0-9]+}}], [[TMP]]
  11   %val = load half, half addrspace(1)* %in
  12   store half %val, half addrspace(1) * %out
  13   ret void
  14 }
  15
  16 define void @test_bitcast_from_half(half addrspace(1)* %in, i16 addrspace(1)* %out) {
  17 ; CHECK-LABEL: @test_bitcast_from_half
  18 ; CHECK: ld.global.b16 [[TMP:%h[0-9]+]], [{{%r[0-9]+}}]
  19 ; CHECK: st.global.b16 [{{%r[0-9]+}}], [[TMP]]
  20   %val = load half, half addrspace(1) * %in
  21   %val_int = bitcast half %val to i16
  22   store i16 %val_int, i16 addrspace(1)* %out
  23   ret void
  24 }
  25
  26 define void @test_bitcast_to_half(half addrspace(1)* %out, i16 addrspace(1)* %in) {
  27 ; CHECK-LABEL: @test_bitcast_to_half
  28 ; CHECK: ld.global.u16 [[TMP:%rs[0-9]+]], [{{%r[0-9]+}}]
  29 ; CHECK: st.global.u16 [{{%r[0-9]+}}], [[TMP]]
  30   %val = load i16, i16 addrspace(1)* %in
  31   %val_fp = bitcast i16 %val to half
  32   store half %val_fp, half addrspace(1)* %out
  33   ret void
  34 }
  35
  36 define void @test_extend32(half addrspace(1)* %in, float addrspace(1)* %out) {
  37 ; CHECK-LABEL: @test_extend32
  38 ; CHECK: cvt.f32.f16
  39
  40   %val16 = load half, half addrspace(1)* %in
  41   %val32 = fpext half %val16 to float
  42   store float %val32, float addrspace(1)* %out
  43   ret void
  44 }
  45
  46 define void @test_extend64(half addrspace(1)* %in, double addrspace(1)* %out) {
  47 ; CHECK-LABEL: @test_extend64
  48 ; CHECK: cvt.f64.f16
  49
  50   %val16 = load half, half addrspace(1)* %in
  51   %val64 = fpext half %val16 to double
  52   store double %val64, double addrspace(1)* %out
  53   ret void
  54 }
  55
  56 define void @test_trunc32(float addrspace(1)* %in, half addrspace(1)* %out) {
  57 ; CHECK-LABEL: test_trunc32
  58 ; CHECK: cvt.rn.f16.f32
  59
  60   %val32 = load float, float addrspace(1)* %in
  61   %val16 = fptrunc float %val32 to half
  62   store half %val16, half addrspace(1)* %out
  63   ret void
  64 }
  65
  66 define void @test_trunc64(double addrspace(1)* %in, half addrspace(1)* %out) {
  67 ; CHECK-LABEL: @test_trunc64
  68 ; CHECK: cvt.rn.f16.f64
  69
  70   %val32 = load double, double addrspace(1)* %in
  71   %val16 = fptrunc double %val32 to half
  72   store half %val16, half addrspace(1)* %out
  73   ret void
  74 }