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
137
138
139
140
141
142
143
|
; RUN: opt -S -scalarizer -dxil-op-lower -mtriple=dxil-pc-shadermodel6.3-library < %s | FileCheck %s
; Test that for scalar values, WaveActiveSum maps down to the DirectX op
define noundef half @wave_active_sum_half(half noundef %expr) {
entry:
; CHECK: call half @dx.op.waveActiveOp.f16(i32 119, half %expr, i8 0, i8 0)
%ret = call half @llvm.dx.wave.reduce.sum.f16(half %expr)
ret half %ret
}
define noundef float @wave_active_sum_float(float noundef %expr) {
entry:
; CHECK: call float @dx.op.waveActiveOp.f32(i32 119, float %expr, i8 0, i8 0)
%ret = call float @llvm.dx.wave.reduce.sum.f32(float %expr)
ret float %ret
}
define noundef double @wave_active_sum_double(double noundef %expr) {
entry:
; CHECK: call double @dx.op.waveActiveOp.f64(i32 119, double %expr, i8 0, i8 0)
%ret = call double @llvm.dx.wave.reduce.sum.f64(double %expr)
ret double %ret
}
define noundef i16 @wave_active_sum_i16(i16 noundef %expr) {
entry:
; CHECK: call i16 @dx.op.waveActiveOp.i16(i32 119, i16 %expr, i8 0, i8 0)
%ret = call i16 @llvm.dx.wave.reduce.sum.i16(i16 %expr)
ret i16 %ret
}
define noundef i32 @wave_active_sum_i32(i32 noundef %expr) {
entry:
; CHECK: call i32 @dx.op.waveActiveOp.i32(i32 119, i32 %expr, i8 0, i8 0)
%ret = call i32 @llvm.dx.wave.reduce.sum.i32(i32 %expr)
ret i32 %ret
}
define noundef i64 @wave_active_sum_i64(i64 noundef %expr) {
entry:
; CHECK: call i64 @dx.op.waveActiveOp.i64(i32 119, i64 %expr, i8 0, i8 0)
%ret = call i64 @llvm.dx.wave.reduce.sum.i64(i64 %expr)
ret i64 %ret
}
define noundef i16 @wave_active_usum_i16(i16 noundef %expr) {
entry:
; CHECK: call i16 @dx.op.waveActiveOp.i16(i32 119, i16 %expr, i8 0, i8 1)
%ret = call i16 @llvm.dx.wave.reduce.usum.i16(i16 %expr)
ret i16 %ret
}
define noundef i32 @wave_active_usum_i32(i32 noundef %expr) {
entry:
; CHECK: call i32 @dx.op.waveActiveOp.i32(i32 119, i32 %expr, i8 0, i8 1)
%ret = call i32 @llvm.dx.wave.reduce.usum.i32(i32 %expr)
ret i32 %ret
}
define noundef i64 @wave_active_usum_i64(i64 noundef %expr) {
entry:
; CHECK: call i64 @dx.op.waveActiveOp.i64(i32 119, i64 %expr, i8 0, i8 1)
%ret = call i64 @llvm.dx.wave.reduce.usum.i64(i64 %expr)
ret i64 %ret
}
declare half @llvm.dx.wave.reduce.sum.f16(half)
declare float @llvm.dx.wave.reduce.sum.f32(float)
declare double @llvm.dx.wave.reduce.sum.f64(double)
declare i16 @llvm.dx.wave.reduce.sum.i16(i16)
declare i32 @llvm.dx.wave.reduce.sum.i32(i32)
declare i64 @llvm.dx.wave.reduce.sum.i64(i64)
declare i16 @llvm.dx.wave.reduce.usum.i16(i16)
declare i32 @llvm.dx.wave.reduce.usum.i32(i32)
declare i64 @llvm.dx.wave.reduce.usum.i64(i64)
; Test that for vector values, WaveActiveSum scalarizes and maps down to the
; DirectX op
define noundef <2 x half> @wave_active_sum_v2half(<2 x half> noundef %expr) {
entry:
; CHECK: call half @dx.op.waveActiveOp.f16(i32 119, half %expr.i0, i8 0, i8 0)
; CHECK: call half @dx.op.waveActiveOp.f16(i32 119, half %expr.i1, i8 0, i8 0)
%ret = call <2 x half> @llvm.dx.wave.reduce.sum.v2f16(<2 x half> %expr)
ret <2 x half> %ret
}
define noundef <3 x i32> @wave_active_sum_v3i32(<3 x i32> noundef %expr) {
entry:
; CHECK: call i32 @dx.op.waveActiveOp.i32(i32 119, i32 %expr.i0, i8 0, i8 0)
; CHECK: call i32 @dx.op.waveActiveOp.i32(i32 119, i32 %expr.i1, i8 0, i8 0)
; CHECK: call i32 @dx.op.waveActiveOp.i32(i32 119, i32 %expr.i2, i8 0, i8 0)
%ret = call <3 x i32> @llvm.dx.wave.reduce.sum.v3i32(<3 x i32> %expr)
ret <3 x i32> %ret
}
define noundef <4 x double> @wave_active_sum_v4f64(<4 x double> noundef %expr) {
entry:
; CHECK: call double @dx.op.waveActiveOp.f64(i32 119, double %expr.i0, i8 0, i8 0)
; CHECK: call double @dx.op.waveActiveOp.f64(i32 119, double %expr.i1, i8 0, i8 0)
; CHECK: call double @dx.op.waveActiveOp.f64(i32 119, double %expr.i2, i8 0, i8 0)
; CHECK: call double @dx.op.waveActiveOp.f64(i32 119, double %expr.i3, i8 0, i8 0)
%ret = call <4 x double> @llvm.dx.wave.reduce.sum.v464(<4 x double> %expr)
ret <4 x double> %ret
}
declare <2 x half> @llvm.dx.wave.reduce.sum.v2f16(<2 x half>)
declare <3 x i32> @llvm.dx.wave.reduce.sum.v3i32(<3 x i32>)
declare <4 x double> @llvm.dx.wave.reduce.sum.v4f64(<4 x double>)
define noundef <2 x i16> @wave_active_usum_v2i16(<2 x i16> noundef %expr) {
entry:
; CHECK: call i16 @dx.op.waveActiveOp.i16(i32 119, i16 %expr.i0, i8 0, i8 1)
; CHECK: call i16 @dx.op.waveActiveOp.i16(i32 119, i16 %expr.i1, i8 0, i8 1)
%ret = call <2 x i16> @llvm.dx.wave.reduce.usum.v2f16(<2 x i16> %expr)
ret <2 x i16> %ret
}
define noundef <3 x i32> @wave_active_usum_v3i32(<3 x i32> noundef %expr) {
entry:
; CHECK: call i32 @dx.op.waveActiveOp.i32(i32 119, i32 %expr.i0, i8 0, i8 1)
; CHECK: call i32 @dx.op.waveActiveOp.i32(i32 119, i32 %expr.i1, i8 0, i8 1)
; CHECK: call i32 @dx.op.waveActiveOp.i32(i32 119, i32 %expr.i2, i8 0, i8 1)
%ret = call <3 x i32> @llvm.dx.wave.reduce.usum.v3i32(<3 x i32> %expr)
ret <3 x i32> %ret
}
define noundef <4 x i64> @wave_active_usum_v4f64(<4 x i64> noundef %expr) {
entry:
; CHECK: call i64 @dx.op.waveActiveOp.i64(i32 119, i64 %expr.i0, i8 0, i8 1)
; CHECK: call i64 @dx.op.waveActiveOp.i64(i32 119, i64 %expr.i1, i8 0, i8 1)
; CHECK: call i64 @dx.op.waveActiveOp.i64(i32 119, i64 %expr.i2, i8 0, i8 1)
; CHECK: call i64 @dx.op.waveActiveOp.i64(i32 119, i64 %expr.i3, i8 0, i8 1)
%ret = call <4 x i64> @llvm.dx.wave.reduce.usum.v464(<4 x i64> %expr)
ret <4 x i64> %ret
}
declare <2 x i16> @llvm.dx.wave.reduce.usum.v2f16(<2 x i16>)
declare <3 x i32> @llvm.dx.wave.reduce.usum.v3i32(<3 x i32>)
declare <4 x i64> @llvm.dx.wave.reduce.usum.v4f64(<4 x i64>)
|
