forked from OSchip/llvm-project
177 lines
5.7 KiB
C
177 lines
5.7 KiB
C
// RUN: %clang_cc1 %s -O3 -triple=x86_64-apple-darwin -target-feature +avx -emit-llvm -o - | FileCheck %s
|
|
|
|
// Don't include mm_malloc.h, it's system specific.
|
|
#define __MM_MALLOC_H
|
|
|
|
#include <immintrin.h>
|
|
|
|
//
|
|
// Test LLVM IR codegen of shuffle instructions
|
|
//
|
|
|
|
__m256 x(__m256 a, __m256 b) {
|
|
// Check if the mask is correct
|
|
// CHECK: shufflevector{{.*}}<i32 3, i32 2, i32 8, i32 11, i32 7, i32 6, i32 12, i32 15>
|
|
return _mm256_shuffle_ps(a, b, 203);
|
|
}
|
|
|
|
__m128d test_mm_permute_pd(__m128d a) {
|
|
// Check if the mask is correct
|
|
// CHECK: shufflevector{{.*}}<i32 1, i32 0>
|
|
return _mm_permute_pd(a, 1);
|
|
}
|
|
|
|
__m256d test_mm256_permute_pd(__m256d a) {
|
|
// Check if the mask is correct
|
|
// CHECK: shufflevector{{.*}}<i32 1, i32 0, i32 3, i32 2>
|
|
return _mm256_permute_pd(a, 5);
|
|
}
|
|
|
|
__m128 test_mm_permute_ps(__m128 a) {
|
|
// Check if the mask is correct
|
|
// CHECK: shufflevector{{.*}}<i32 3, i32 2, i32 1, i32 0>
|
|
return _mm_permute_ps(a, 0x1b);
|
|
}
|
|
|
|
// Test case for PR12401
|
|
__m128 test_mm_permute_ps2(__m128 a) {
|
|
// Check if the mask is correct
|
|
// CHECK: shufflevector{{.*}}<i32 2, i32 1, i32 2, i32 3>
|
|
return _mm_permute_ps(a, 0xe6);
|
|
}
|
|
|
|
__m256 test_mm256_permute_ps(__m256 a) {
|
|
// Check if the mask is correct
|
|
// CHECK: shufflevector{{.*}}<i32 3, i32 2, i32 1, i32 0, i32 7, i32 6, i32 5, i32 4>
|
|
return _mm256_permute_ps(a, 0x1b);
|
|
}
|
|
|
|
__m256d test_mm256_permute2f128_pd(__m256d a, __m256d b) {
|
|
// Check if the mask is correct
|
|
// CHECK: shufflevector{{.*}}<i32 2, i32 3, i32 6, i32 7>
|
|
return _mm256_permute2f128_pd(a, b, 0x31);
|
|
}
|
|
|
|
__m256 test_mm256_permute2f128_ps(__m256 a, __m256 b) {
|
|
// Check if the mask is correct
|
|
// CHECK: shufflevector{{.*}}<i32 4, i32 5, i32 6, i32 7, i32 12, i32 13, i32 14, i32 15>
|
|
return _mm256_permute2f128_ps(a, b, 0x13);
|
|
}
|
|
|
|
__m256i test_mm256_permute2f128_si256(__m256i a, __m256i b) {
|
|
// Check if the mask is correct
|
|
// CHECK: shufflevector{{.*}} <8 x i32> <i32 0, i32 1, i32 2, i32 3, i32 8, i32 9, i32 10, i32 11>
|
|
return _mm256_permute2f128_si256(a, b, 0x20);
|
|
}
|
|
|
|
__m128
|
|
test_mm_broadcast_ss(float const *__a) {
|
|
// CHECK-LABEL: @test_mm_broadcast_ss
|
|
// CHECK: insertelement <4 x float> {{.*}}, i32 0
|
|
// CHECK: insertelement <4 x float> {{.*}}, i32 1
|
|
// CHECK: insertelement <4 x float> {{.*}}, i32 2
|
|
// CHECK: insertelement <4 x float> {{.*}}, i32 3
|
|
return _mm_broadcast_ss(__a);
|
|
}
|
|
|
|
__m256d
|
|
test_mm256_broadcast_sd(double const *__a) {
|
|
// CHECK-LABEL: @test_mm256_broadcast_sd
|
|
// CHECK: insertelement <4 x double> {{.*}}, i32 0
|
|
// CHECK: insertelement <4 x double> {{.*}}, i32 1
|
|
// CHECK: insertelement <4 x double> {{.*}}, i32 2
|
|
// CHECK: insertelement <4 x double> {{.*}}, i32 3
|
|
return _mm256_broadcast_sd(__a);
|
|
}
|
|
|
|
__m256
|
|
test_mm256_broadcast_ss(float const *__a) {
|
|
// CHECK-LABEL: @test_mm256_broadcast_ss
|
|
// CHECK: insertelement <8 x float> {{.*}}, i32 0
|
|
// CHECK: insertelement <8 x float> {{.*}}, i32 1
|
|
// CHECK: insertelement <8 x float> {{.*}}, i32 2
|
|
// CHECK: insertelement <8 x float> {{.*}}, i32 3
|
|
// CHECK: insertelement <8 x float> {{.*}}, i32 4
|
|
// CHECK: insertelement <8 x float> {{.*}}, i32 5
|
|
// CHECK: insertelement <8 x float> {{.*}}, i32 6
|
|
// CHECK: insertelement <8 x float> {{.*}}, i32 7
|
|
return _mm256_broadcast_ss(__a);
|
|
}
|
|
|
|
// Make sure we have the correct mask for each insertf128 case.
|
|
|
|
__m256 test_mm256_insertf128_ps_0(__m256 a, __m128 b) {
|
|
// CHECK-LABEL: @test_mm256_insertf128_ps_0
|
|
// CHECK: shufflevector{{.*}}<i32 8, i32 9, i32 10, i32 11, i32 4, i32 5, i32 6, i32 7>
|
|
return _mm256_insertf128_ps(a, b, 0);
|
|
}
|
|
|
|
__m256d test_mm256_insertf128_pd_0(__m256d a, __m128d b) {
|
|
// CHECK-LABEL: @test_mm256_insertf128_pd_0
|
|
// CHECK: shufflevector{{.*}}<i32 4, i32 5, i32 2, i32 3>
|
|
return _mm256_insertf128_pd(a, b, 0);
|
|
}
|
|
|
|
__m256i test_mm256_insertf128_si256_0(__m256i a, __m128i b) {
|
|
// CHECK-LABEL: @test_mm256_insertf128_si256_0
|
|
// CHECK: shufflevector{{.*}}<i32 4, i32 5, i32 2, i32 3>
|
|
return _mm256_insertf128_si256(a, b, 0);
|
|
}
|
|
|
|
__m256 test_mm256_insertf128_ps_1(__m256 a, __m128 b) {
|
|
// CHECK-LABEL: @test_mm256_insertf128_ps_1
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3, i32 8, i32 9, i32 10, i32 11>
|
|
return _mm256_insertf128_ps(a, b, 1);
|
|
}
|
|
|
|
__m256d test_mm256_insertf128_pd_1(__m256d a, __m128d b) {
|
|
// CHECK-LABEL: @test_mm256_insertf128_pd_1
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 4, i32 5>
|
|
return _mm256_insertf128_pd(a, b, 1);
|
|
}
|
|
|
|
__m256i test_mm256_insertf128_si256_1(__m256i a, __m128i b) {
|
|
// CHECK-LABEL: @test_mm256_insertf128_si256_1
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 4, i32 5>
|
|
return _mm256_insertf128_si256(a, b, 1);
|
|
}
|
|
|
|
// Make sure we have the correct mask for each extractf128 case.
|
|
|
|
__m128 test_mm256_extractf128_ps_0(__m256 a) {
|
|
// CHECK-LABEL: @test_mm256_extractf128_ps_0
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>
|
|
return _mm256_extractf128_ps(a, 0);
|
|
}
|
|
|
|
__m128d test_mm256_extractf128_pd_0(__m256d a) {
|
|
// CHECK-LABEL: @test_mm256_extractf128_pd_0
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1>
|
|
return _mm256_extractf128_pd(a, 0);
|
|
}
|
|
|
|
__m128i test_mm256_extractf128_si256_0(__m256i a) {
|
|
// CHECK-LABEL: @test_mm256_extractf128_si256_0
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1>
|
|
return _mm256_extractf128_si256(a, 0);
|
|
}
|
|
|
|
__m128 test_mm256_extractf128_ps_1(__m256 a) {
|
|
// CHECK-LABEL: @test_mm256_extractf128_ps_1
|
|
// CHECK: shufflevector{{.*}}<i32 4, i32 5, i32 6, i32 7>
|
|
return _mm256_extractf128_ps(a, 1);
|
|
}
|
|
|
|
__m128d test_mm256_extractf128_pd_1(__m256d a) {
|
|
// CHECK-LABEL: @test_mm256_extractf128_pd_1
|
|
// CHECK: shufflevector{{.*}}<i32 2, i32 3>
|
|
return _mm256_extractf128_pd(a, 1);
|
|
}
|
|
|
|
__m128i test_mm256_extractf128_si256_1(__m256i a) {
|
|
// CHECK-LABEL: @test_mm256_extractf128_si256_1
|
|
// CHECK: shufflevector{{.*}}<i32 2, i32 3>
|
|
return _mm256_extractf128_si256(a, 1);
|
|
}
|
|
|