llvm/test/CodeGen/NVPTX/convert-sm80.ll

   1 ; RUN: llc < %s -march=nvptx64 -mcpu=sm_80 -mattr=+ptx70 | FileCheck %s
   2 ; RUN: %if ptxas-11.0 %{ llc < %s -march=nvptx64 -mcpu=sm_80 -mattr=+ptx70 | %ptxas-verify -arch=sm_80 %}
   3
   4
   5 ; CHECK-LABEL: cvt_rn_bf16x2_f32
   6 define <2 x bfloat> @cvt_rn_bf16x2_f32(float %f1, float %f2) {
   7
   8 ; CHECK: cvt.rn.bf16x2.f32
   9   %val = call <2 x bfloat> @llvm.nvvm.ff2bf16x2.rn(float %f1, float %f2);
  10
  11 ret <2 x bfloat> %val
  12 }
  13
  14 ; CHECK-LABEL: cvt_rn_relu_bf16x2_f32
  15 define <2 x bfloat> @cvt_rn_relu_bf16x2_f32(float %f1, float %f2) {
  16
  17 ; CHECK: cvt.rn.relu.bf16x2.f32
  18 %val = call <2 x bfloat> @llvm.nvvm.ff2bf16x2.rn.relu(float %f1, float %f2);
  19
  20 ret <2 x bfloat> %val
  21 }
  22
  23 ; CHECK-LABEL: cvt_rz_bf16x2_f32
  24 define <2 x bfloat> @cvt_rz_bf16x2_f32(float %f1, float %f2) {
  25
  26 ; CHECK: cvt.rz.bf16x2.f32
  27   %val = call <2 x bfloat> @llvm.nvvm.ff2bf16x2.rz(float %f1, float %f2);
  28
  29 ret <2 x bfloat> %val
  30 }
  31
  32 ; CHECK-LABEL: cvt_rz_relu_bf16x2_f32
  33 define <2 x bfloat> @cvt_rz_relu_bf16x2_f32(float %f1, float %f2) {
  34
  35 ; CHECK: cvt.rz.relu.bf16x2.f32
  36 %val = call <2 x bfloat> @llvm.nvvm.ff2bf16x2.rz.relu(float %f1, float %f2);
  37
  38 ret <2 x bfloat> %val
  39 }
  40
  41 declare <2 x bfloat> @llvm.nvvm.ff2bf16x2.rn(float, float)
  42 declare <2 x bfloat> @llvm.nvvm.ff2bf16x2.rn.relu(float, float)
  43 declare <2 x bfloat> @llvm.nvvm.ff2bf16x2.rz(float, float)
  44 declare <2 x bfloat> @llvm.nvvm.ff2bf16x2.rz.relu(float, float)
  45
  46 ; CHECK-LABEL: cvt_rn_f16x2_f32
  47 define <2 x half> @cvt_rn_f16x2_f32(float %f1, float %f2) {
  48
  49 ; CHECK: cvt.rn.f16x2.f32
  50   %val = call <2 x half> @llvm.nvvm.ff2f16x2.rn(float %f1, float %f2);
  51
  52 ret <2 x half> %val
  53 }
  54
  55 ; CHECK-LABEL: cvt_rn_relu_f16x2_f32
  56 define <2 x half> @cvt_rn_relu_f16x2_f32(float %f1, float %f2) {
  57
  58 ; CHECK: cvt.rn.relu.f16x2.f32
  59 %val = call <2 x half> @llvm.nvvm.ff2f16x2.rn.relu(float %f1, float %f2);
  60
  61 ret <2 x half> %val
  62 }
  63
  64 ; CHECK-LABEL: cvt_rz_f16x2_f32
  65 define <2 x half> @cvt_rz_f16x2_f32(float %f1, float %f2) {
  66
  67 ; CHECK: cvt.rz.f16x2.f32
  68   %val = call <2 x half> @llvm.nvvm.ff2f16x2.rz(float %f1, float %f2);
  69
  70 ret <2 x half> %val
  71 }
  72
  73 ; CHECK-LABEL: cvt_rz_relu_f16x2_f32
  74 define <2 x half> @cvt_rz_relu_f16x2_f32(float %f1, float %f2) {
  75
  76 ; CHECK: cvt.rz.relu.f16x2.f32
  77 %val = call <2 x half> @llvm.nvvm.ff2f16x2.rz.relu(float %f1, float %f2);
  78
  79 ret <2 x half> %val
  80 }
  81
  82 declare <2 x half> @llvm.nvvm.ff2f16x2.rn(float, float)
  83 declare <2 x half> @llvm.nvvm.ff2f16x2.rn.relu(float, float)
  84 declare <2 x half> @llvm.nvvm.ff2f16x2.rz(float, float)
  85 declare <2 x half> @llvm.nvvm.ff2f16x2.rz.relu(float, float)
  86
  87 ; CHECK-LABEL: cvt_rn_bf16_f32
  88 define bfloat @cvt_rn_bf16_f32(float %f1) {
  89
  90 ; CHECK: cvt.rn.bf16.f32
  91   %val = call bfloat @llvm.nvvm.f2bf16.rn(float %f1);
  92
  93 ret bfloat %val
  94 }
  95
  96 ; CHECK-LABEL: cvt_rn_relu_bf16_f32
  97 define bfloat @cvt_rn_relu_bf16_f32(float %f1) {
  98
  99 ; CHECK: cvt.rn.relu.bf16.f32
 100 %val = call bfloat @llvm.nvvm.f2bf16.rn.relu(float %f1);
 101
 102 ret bfloat %val
 103 }
 104
 105 ; CHECK-LABEL: cvt_rz_bf16_f32
 106 define bfloat @cvt_rz_bf16_f32(float %f1) {
 107
 108 ; CHECK: cvt.rz.bf16.f32
 109   %val = call bfloat @llvm.nvvm.f2bf16.rz(float %f1);
 110
 111 ret bfloat %val
 112 }
 113
 114 ; CHECK-LABEL: cvt_rz_relu_bf16_f32
 115 define bfloat @cvt_rz_relu_bf16_f32(float %f1) {
 116
 117 ; CHECK: cvt.rz.relu.bf16.f32
 118 %val = call bfloat @llvm.nvvm.f2bf16.rz.relu(float %f1);
 119
 120 ret bfloat %val
 121 }
 122
 123 declare bfloat @llvm.nvvm.f2bf16.rn(float)
 124 declare bfloat @llvm.nvvm.f2bf16.rn.relu(float)
 125 declare bfloat @llvm.nvvm.f2bf16.rz(float)
 126 declare bfloat @llvm.nvvm.f2bf16.rz.relu(float)
 127
 128 ; CHECK-LABEL: cvt_rna_tf32_f32
 129 define i32 @cvt_rna_tf32_f32(float %f1) {
 130
 131 ; CHECK: cvt.rna.tf32.f32
 132   %val = call i32 @llvm.nvvm.f2tf32.rna(float %f1);
 133
 134 ret i32 %val
 135 }
 136
 137 declare i32 @llvm.nvvm.f2tf32.rna(float)