xref: /llvm-project/llvm/test/CodeGen/DirectX/firstbithigh.ll (revision 011b618644113996e2c0a8e57db40f89d20878e3)
1; RUN: opt -S -scalarizer -dxil-op-lower -mtriple=dxil-pc-shadermodel6.3-library %s | FileCheck %s
2
3; Make sure dxil operation function calls for firstbithigh are generated for all integer types.
4
5define noundef i32 @test_firstbithigh_ushort(i16 noundef %a) {
6entry:
7; CHECK: call i32 @dx.op.unaryBits.i16(i32 33, i16 %{{.*}}) #[[#ATTR:]]
8  %elt.firstbithigh = call i32 @llvm.dx.firstbituhigh.i16(i16 %a)
9  ret i32 %elt.firstbithigh
10}
11
12define noundef i32 @test_firstbithigh_short(i16 noundef %a) {
13entry:
14; CHECK: call i32 @dx.op.unaryBits.i16(i32 34, i16 %{{.*}}) #[[#ATTR]]
15  %elt.firstbithigh = call i32 @llvm.dx.firstbitshigh.i16(i16 %a)
16  ret i32 %elt.firstbithigh
17}
18
19define noundef i32 @test_firstbithigh_uint(i32 noundef %a) {
20entry:
21; CHECK: call i32 @dx.op.unaryBits.i32(i32 33, i32 %{{.*}}) #[[#ATTR]]
22  %elt.firstbithigh = call i32 @llvm.dx.firstbituhigh.i32(i32 %a)
23  ret i32 %elt.firstbithigh
24}
25
26define noundef i32 @test_firstbithigh_int(i32 noundef %a) {
27entry:
28; CHECK: call i32 @dx.op.unaryBits.i32(i32 34, i32 %{{.*}}) #[[#ATTR]]
29  %elt.firstbithigh = call i32 @llvm.dx.firstbitshigh.i32(i32 %a)
30  ret i32 %elt.firstbithigh
31}
32
33define noundef i32 @test_firstbithigh_ulong(i64 noundef %a) {
34entry:
35; CHECK: call i32 @dx.op.unaryBits.i64(i32 33, i64 %{{.*}}) #[[#ATTR]]
36  %elt.firstbithigh = call i32 @llvm.dx.firstbituhigh.i64(i64 %a)
37  ret i32 %elt.firstbithigh
38}
39
40define noundef i32 @test_firstbithigh_long(i64 noundef %a) {
41entry:
42; CHECK: call i32 @dx.op.unaryBits.i64(i32 34, i64 %{{.*}}) #[[#ATTR]]
43  %elt.firstbithigh = call i32 @llvm.dx.firstbitshigh.i64(i64 %a)
44  ret i32 %elt.firstbithigh
45}
46
47define noundef <4 x i32> @test_firstbituhigh_vec4_i32(<4 x i32> noundef %a)  {
48entry:
49  ; CHECK: [[ee0:%.*]] = extractelement <4 x i32> %a, i64 0
50  ; CHECK: [[ie0:%.*]] = call i32 @dx.op.unaryBits.i32(i32 33, i32 [[ee0]]) #[[#ATTR]]
51  ; CHECK: [[ee1:%.*]] = extractelement <4 x i32> %a, i64 1
52  ; CHECK: [[ie1:%.*]] = call i32 @dx.op.unaryBits.i32(i32 33, i32 [[ee1]]) #[[#ATTR]]
53  ; CHECK: [[ee2:%.*]] = extractelement <4 x i32> %a, i64 2
54  ; CHECK: [[ie2:%.*]] = call i32 @dx.op.unaryBits.i32(i32 33, i32 [[ee2]]) #[[#ATTR]]
55  ; CHECK: [[ee3:%.*]] = extractelement <4 x i32> %a, i64 3
56  ; CHECK: [[ie3:%.*]] = call i32 @dx.op.unaryBits.i32(i32 33, i32 [[ee3]]) #[[#ATTR]]
57  ; CHECK: insertelement <4 x i32> poison, i32 [[ie0]], i64 0
58  ; CHECK: insertelement <4 x i32> %{{.*}}, i32 [[ie1]], i64 1
59  ; CHECK: insertelement <4 x i32> %{{.*}}, i32 [[ie2]], i64 2
60  ; CHECK: insertelement <4 x i32> %{{.*}}, i32 [[ie3]], i64 3
61  %2 = call <4 x i32> @llvm.dx.firstbituhigh.v4i32(<4 x i32> %a)
62  ret <4 x i32> %2
63}
64
65define noundef <4 x i32> @test_firstbitshigh_vec4_i32(<4 x i32> noundef %a)  {
66entry:
67  ; CHECK: [[ee0:%.*]] = extractelement <4 x i32> %a, i64 0
68  ; CHECK: [[ie0:%.*]] = call i32 @dx.op.unaryBits.i32(i32 34, i32 [[ee0]]) #[[#ATTR]]
69  ; CHECK: [[ee1:%.*]] = extractelement <4 x i32> %a, i64 1
70  ; CHECK: [[ie1:%.*]] = call i32 @dx.op.unaryBits.i32(i32 34, i32 [[ee1]]) #[[#ATTR]]
71  ; CHECK: [[ee2:%.*]] = extractelement <4 x i32> %a, i64 2
72  ; CHECK: [[ie2:%.*]] = call i32 @dx.op.unaryBits.i32(i32 34, i32 [[ee2]]) #[[#ATTR]]
73  ; CHECK: [[ee3:%.*]] = extractelement <4 x i32> %a, i64 3
74  ; CHECK: [[ie3:%.*]] = call i32 @dx.op.unaryBits.i32(i32 34, i32 [[ee3]]) #[[#ATTR]]
75  ; CHECK: insertelement <4 x i32> poison, i32 [[ie0]], i64 0
76  ; CHECK: insertelement <4 x i32> %{{.*}}, i32 [[ie1]], i64 1
77  ; CHECK: insertelement <4 x i32> %{{.*}}, i32 [[ie2]], i64 2
78  ; CHECK: insertelement <4 x i32> %{{.*}}, i32 [[ie3]], i64 3
79  %2 = call <4 x i32> @llvm.dx.firstbitshigh.v4i32(<4 x i32> %a)
80  ret <4 x i32> %2
81}
82
83; CHECK: attributes #[[#ATTR]] = {{{.*}} memory(none) {{.*}}}
84
85declare i32 @llvm.dx.firstbituhigh.i16(i16)
86declare i32 @llvm.dx.firstbituhigh.i32(i32)
87declare i32 @llvm.dx.firstbituhigh.i64(i64)
88declare <4 x i32> @llvm.dx.firstbituhigh.v4i32(<4 x i32>)
89
90declare i32 @llvm.dx.firstbitshigh.i16(i16)
91declare i32 @llvm.dx.firstbitshigh.i32(i32)
92declare i32 @llvm.dx.firstbitshigh.i64(i64)
93declare <4 x i32> @llvm.dx.firstbitshigh.v4i32(<4 x i32>)
94