Clang Project

clang_source_code/test/CodeGen/avx-shuffle-builtins.c
1// RUN: %clang_cc1 -ffreestanding %s -O3 -triple=x86_64-apple-darwin -target-feature +avx -emit-llvm -o - | FileCheck %s
2// FIXME: This is testing optimized generation of shuffle instructions and should be fixed.
3
4
5#include <immintrin.h>
6
7//
8// Test LLVM IR codegen of shuffle instructions
9//
10
11__m256 x(__m256 a, __m256 b) {
12  // Check if the mask is correct
13  // CHECK: shufflevector{{.*}}<i32 3, i32 2, i32 8, i32 11, i32 7, i32 6, i32 12, i32 15>
14  return _mm256_shuffle_ps(a, b, 203);
15}
16
17__m128d test_mm_permute_pd(__m128d a) {
18  // Check if the mask is correct
19  // CHECK: shufflevector{{.*}}<i32 1, i32 0>
20  return _mm_permute_pd(a, 1);
21}
22
23__m256d test_mm256_permute_pd(__m256d a) {
24  // Check if the mask is correct
25  // CHECK: shufflevector{{.*}}<i32 1, i32 0, i32 3, i32 2>
26  return _mm256_permute_pd(a, 5);
27}
28
29__m128 test_mm_permute_ps(__m128 a) {
30  // Check if the mask is correct
31  // CHECK: shufflevector{{.*}}<i32 3, i32 2, i32 1, i32 0>
32  return _mm_permute_ps(a, 0x1b);
33}
34
35// Test case for PR12401
36__m128 test_mm_permute_ps2(__m128 a) {
37  // Check if the mask is correct
38  // CHECK: shufflevector{{.*}}<i32 2, i32 1, i32 2, i32 3>
39  return _mm_permute_ps(a, 0xe6);
40}
41
42__m256 test_mm256_permute_ps(__m256 a) {
43  // Check if the mask is correct
44  // CHECK: shufflevector{{.*}}<i32 3, i32 2, i32 1, i32 0, i32 7, i32 6, i32 5, i32 4>
45  return _mm256_permute_ps(a, 0x1b);
46}
47
48__m256d test_mm256_permute2f128_pd(__m256d a, __m256d b) {
49  // Check if the mask is correct
50  // CHECK: shufflevector{{.*}}<i32 2, i32 3, i32 6, i32 7> 
51  return _mm256_permute2f128_pd(a, b, 0x31);
52}
53
54__m256 test_mm256_permute2f128_ps(__m256 a, __m256 b) {
55  // Check if the mask is correct
56  // CHECK: shufflevector{{.*}}<i32 4, i32 5, i32 6, i32 7, i32 12, i32 13, i32 14, i32 15>
57  return _mm256_permute2f128_ps(a, b, 0x13);
58}
59
60__m256i test_mm256_permute2f128_si256(__m256i a, __m256i b) {
61  // Check if the mask is correct
62  // CHECK: shufflevector{{.*}} <8 x i32> <i32 0, i32 1, i32 2, i32 3, i32 8, i32 9, i32 10, i32 11>
63  return _mm256_permute2f128_si256(a, b, 0x20);
64}
65
66__m128
67test_mm_broadcast_ss(float const *__a) {
68  // CHECK-LABEL: @test_mm_broadcast_ss
69  // CHECK: insertelement <4 x float> {{.*}}, i32 0
70  // CHECK: shufflevector <4 x float> {{.*}}, <4 x float> undef, <4 x i32> zeroinitializer
71  return _mm_broadcast_ss(__a);
72}
73
74__m256d
75test_mm256_broadcast_sd(double const *__a) {
76  // CHECK-LABEL: @test_mm256_broadcast_sd
77  // CHECK: insertelement <4 x double> {{.*}}, i32 0
78  // CHECK: shufflevector <4 x double> {{.*}}, <4 x double> undef, <4 x i32> zeroinitializer
79  return _mm256_broadcast_sd(__a);
80}
81
82__m256
83test_mm256_broadcast_ss(float const *__a) {
84  // CHECK-LABEL: @test_mm256_broadcast_ss
85  // CHECK: insertelement <8 x float> {{.*}}, i32 0
86  // CHECK: shufflevector <8 x float> {{.*}}, <8 x float> undef, <8 x i32> zeroinitializer
87  return _mm256_broadcast_ss(__a);
88}
89
90// Make sure we have the correct mask for each insertf128 case.
91
92__m256 test_mm256_insertf128_ps_0(__m256 a, __m128 b) {
93  // CHECK-LABEL: @test_mm256_insertf128_ps_0
94  // CHECK: shufflevector{{.*}}<i32 8, i32 9, i32 10, i32 11, i32 4, i32 5, i32 6, i32 7>
95  return _mm256_insertf128_ps(a, b, 0);
96}
97
98__m256d test_mm256_insertf128_pd_0(__m256d a, __m128d b) {
99  // CHECK-LABEL: @test_mm256_insertf128_pd_0
100  // CHECK: shufflevector{{.*}}<i32 4, i32 5, i32 2, i32 3>
101  return _mm256_insertf128_pd(a, b, 0);
102}
103
104__m256i test_mm256_insertf128_si256_0(__m256i a, __m128i b) {
105  // CHECK-LABEL: @test_mm256_insertf128_si256_0
106  // CHECK: shufflevector{{.*}}<i32 8, i32 9, i32 10, i32 11, i32 4, i32 5, i32 6, i32 7>
107  return _mm256_insertf128_si256(a, b, 0);
108}
109
110__m256 test_mm256_insertf128_ps_1(__m256 a, __m128 b) {
111  // CHECK-LABEL: @test_mm256_insertf128_ps_1
112  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3, i32 8, i32 9, i32 10, i32 11>
113  return _mm256_insertf128_ps(a, b, 1);
114}
115
116__m256d test_mm256_insertf128_pd_1(__m256d a, __m128d b) {
117  // CHECK-LABEL: @test_mm256_insertf128_pd_1
118  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 4, i32 5>
119  return _mm256_insertf128_pd(a, b, 1);
120}
121
122__m256i test_mm256_insertf128_si256_1(__m256i a, __m128i b) {
123  // CHECK-LABEL: @test_mm256_insertf128_si256_1
124  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3, i32 8, i32 9, i32 10, i32 11>
125  return _mm256_insertf128_si256(a, b, 1);
126}
127
128// Make sure we have the correct mask for each extractf128 case.
129
130__m128 test_mm256_extractf128_ps_0(__m256 a) {
131  // CHECK-LABEL: @test_mm256_extractf128_ps_0
132  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>
133  return _mm256_extractf128_ps(a, 0);
134}
135
136__m128d test_mm256_extractf128_pd_0(__m256d a) {
137  // CHECK-LABEL: @test_mm256_extractf128_pd_0
138  // CHECK: shufflevector{{.*}}<i32 0, i32 1>
139  return _mm256_extractf128_pd(a, 0);
140}
141
142__m128i test_mm256_extractf128_si256_0(__m256i a) {
143  // CHECK-LABEL: @test_mm256_extractf128_si256_0
144  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>
145  return _mm256_extractf128_si256(a, 0);
146}
147
148__m128 test_mm256_extractf128_ps_1(__m256 a) {
149  // CHECK-LABEL: @test_mm256_extractf128_ps_1
150  // CHECK: shufflevector{{.*}}<i32 4, i32 5, i32 6, i32 7>
151  return _mm256_extractf128_ps(a, 1);
152}
153
154__m128d test_mm256_extractf128_pd_1(__m256d a) {
155  // CHECK-LABEL: @test_mm256_extractf128_pd_1
156  // CHECK: shufflevector{{.*}}<i32 2, i32 3>
157  return _mm256_extractf128_pd(a, 1);
158}
159
160__m128i test_mm256_extractf128_si256_1(__m256i a) {
161  // CHECK-LABEL: @test_mm256_extractf128_si256_1
162  // CHECK: shufflevector{{.*}}<i32 4, i32 5, i32 6, i32 7>
163  return _mm256_extractf128_si256(a, 1);
164}
165
166__m256 test_mm256_set_m128(__m128 hi, __m128 lo) {
167  // CHECK-LABEL: @test_mm256_set_m128
168  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7>
169  return _mm256_set_m128(hi, lo);
170}
171
172__m256d test_mm256_set_m128d(__m128d hi, __m128d lo) {
173  // CHECK-LABEL: @test_mm256_set_m128d
174  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>
175  return _mm256_set_m128d(hi, lo);
176}
177
178__m256i test_mm256_set_m128i(__m128i hi, __m128i lo) {
179  // CHECK-LABEL: @test_mm256_set_m128i
180  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>  
181  return _mm256_set_m128i(hi, lo);
182}
183
184__m256 test_mm256_setr_m128(__m128 hi, __m128 lo) {
185  // CHECK-LABEL: @test_mm256_setr_m128
186  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7>
187  return _mm256_setr_m128(lo, hi);
188}
189
190__m256d test_mm256_setr_m128d(__m128d hi, __m128d lo) {
191  // CHECK-LABEL: @test_mm256_setr_m128d
192  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>
193  return _mm256_setr_m128d(lo, hi);
194}
195
196__m256i test_mm256_setr_m128i(__m128i hi, __m128i lo) {
197  // CHECK-LABEL: @test_mm256_setr_m128i
198  // CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>
199  return _mm256_setr_m128i(lo, hi);
200}
201
CodeThinker, Enjoy your reading code