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