mirror of
https://github.com/Gericom/teak-llvm.git
synced 2025-06-19 19:45:40 -04:00
b276dd195a
In PR41304: https://bugs.llvm.org/show_bug.cgi?id=41304 ...we have a case where we want to fold a binop of select-shuffle (blended) values. Rather than try to match commuted variants of the pattern, we can canonicalize the shuffles and check for mask equality with commuted operands. We don't produce arbitrary shuffle masks in instcombine, but select-shuffles are a special case that the backend is required to handle because we already canonicalize vector select to this shuffle form. So there should be no codegen difference from this change. It's possible that this improves CSE in IR though. Differential Revision: https://reviews.llvm.org/D60016 llvm-svn: 357366
201 lines
6.6 KiB
C
201 lines
6.6 KiB
C
// RUN: %clang_cc1 -ffreestanding %s -O3 -triple=x86_64-apple-darwin -target-feature +avx -emit-llvm -o - | FileCheck %s
|
|
// FIXME: This is testing optimized generation of shuffle instructions and should be fixed.
|
|
|
|
|
|
#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: shufflevector <4 x float> {{.*}}, <4 x float> undef, <4 x i32> zeroinitializer
|
|
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: shufflevector <4 x double> {{.*}}, <4 x double> undef, <4 x i32> zeroinitializer
|
|
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: shufflevector <8 x float> {{.*}}, <8 x float> undef, <8 x i32> zeroinitializer
|
|
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 0, i32 1, i32 2, i32 3, i32 12, i32 13, i32 14, i32 15>
|
|
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 0, i32 1, i32 6, i32 7>
|
|
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 0, i32 1, i32 2, i32 3, i32 12, i32 13, i32 14, i32 15>
|
|
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 2, i32 3, i32 8, i32 9, i32 10, i32 11>
|
|
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, i32 2, i32 3>
|
|
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 4, i32 5, i32 6, i32 7>
|
|
return _mm256_extractf128_si256(a, 1);
|
|
}
|
|
|
|
__m256 test_mm256_set_m128(__m128 hi, __m128 lo) {
|
|
// CHECK-LABEL: @test_mm256_set_m128
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7>
|
|
return _mm256_set_m128(hi, lo);
|
|
}
|
|
|
|
__m256d test_mm256_set_m128d(__m128d hi, __m128d lo) {
|
|
// CHECK-LABEL: @test_mm256_set_m128d
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>
|
|
return _mm256_set_m128d(hi, lo);
|
|
}
|
|
|
|
__m256i test_mm256_set_m128i(__m128i hi, __m128i lo) {
|
|
// CHECK-LABEL: @test_mm256_set_m128i
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>
|
|
return _mm256_set_m128i(hi, lo);
|
|
}
|
|
|
|
__m256 test_mm256_setr_m128(__m128 hi, __m128 lo) {
|
|
// CHECK-LABEL: @test_mm256_setr_m128
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7>
|
|
return _mm256_setr_m128(lo, hi);
|
|
}
|
|
|
|
__m256d test_mm256_setr_m128d(__m128d hi, __m128d lo) {
|
|
// CHECK-LABEL: @test_mm256_setr_m128d
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>
|
|
return _mm256_setr_m128d(lo, hi);
|
|
}
|
|
|
|
__m256i test_mm256_setr_m128i(__m128i hi, __m128i lo) {
|
|
// CHECK-LABEL: @test_mm256_setr_m128i
|
|
// CHECK: shufflevector{{.*}}<i32 0, i32 1, i32 2, i32 3>
|
|
return _mm256_setr_m128i(lo, hi);
|
|
}
|