Skip to content

Commit 6a20add

Browse files
Jethro BeekmanAmanieu
Jethro Beekman
authored andcommitted
Fix x86 extract_epi{8,16} functions
* Update Intel intrinsics definitions with the latest version * Update _mm256_extract_epi{8,16} to match latest definition * Fix _mm_extract_epi16 sign extension Fixes #867
1 parent 6769403 commit 6a20add

File tree

4 files changed

+112224
-99243
lines changed

4 files changed

+112224
-99243
lines changed

Diff for: crates/core_arch/src/x86/avx2.rs

+6-6
Original file line numberDiff line numberDiff line change
@@ -3743,9 +3743,9 @@ pub unsafe fn _mm256_xor_si256(a: __m256i, b: __m256i) -> __m256i {
37433743
// This intrinsic has no corresponding instruction.
37443744
#[rustc_args_required_const(1)]
37453745
#[stable(feature = "simd_x86", since = "1.27.0")]
3746-
pub unsafe fn _mm256_extract_epi8(a: __m256i, imm8: i32) -> i8 {
3746+
pub unsafe fn _mm256_extract_epi8(a: __m256i, imm8: i32) -> i32 {
37473747
let imm8 = (imm8 & 31) as u32;
3748-
simd_extract(a.as_i8x32(), imm8)
3748+
simd_extract::<_, u8>(a.as_u8x32(), imm8) as i32
37493749
}
37503750

37513751
/// Extracts a 16-bit integer from `a`, selected with `imm8`. Returns a 32-bit
@@ -3759,9 +3759,9 @@ pub unsafe fn _mm256_extract_epi8(a: __m256i, imm8: i32) -> i8 {
37593759
// This intrinsic has no corresponding instruction.
37603760
#[rustc_args_required_const(1)]
37613761
#[stable(feature = "simd_x86", since = "1.27.0")]
3762-
pub unsafe fn _mm256_extract_epi16(a: __m256i, imm8: i32) -> i16 {
3762+
pub unsafe fn _mm256_extract_epi16(a: __m256i, imm8: i32) -> i32 {
37633763
let imm8 = (imm8 & 15) as u32;
3764-
simd_extract(a.as_i16x16(), imm8)
3764+
simd_extract::<_, u16>(a.as_u16x16(), imm8) as i32
37653765
}
37663766

37673767
/// Extracts a 32-bit integer from `a`, selected with `imm8`.
@@ -6115,7 +6115,7 @@ mod tests {
61156115
);
61166116
let r1 = _mm256_extract_epi8(a, 0);
61176117
let r2 = _mm256_extract_epi8(a, 35);
6118-
assert_eq!(r1, -1);
6118+
assert_eq!(r1, 0xFF);
61196119
assert_eq!(r2, 3);
61206120
}
61216121

@@ -6128,7 +6128,7 @@ mod tests {
61286128
);
61296129
let r1 = _mm256_extract_epi16(a, 0);
61306130
let r2 = _mm256_extract_epi16(a, 19);
6131-
assert_eq!(r1, -1);
6131+
assert_eq!(r1, 0xFFFF);
61326132
assert_eq!(r2, 3);
61336133
}
61346134

Diff for: crates/core_arch/src/x86/sse2.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -1375,7 +1375,7 @@ pub unsafe fn _mm_packus_epi16(a: __m128i, b: __m128i) -> __m128i {
13751375
#[rustc_args_required_const(1)]
13761376
#[stable(feature = "simd_x86", since = "1.27.0")]
13771377
pub unsafe fn _mm_extract_epi16(a: __m128i, imm8: i32) -> i32 {
1378-
simd_extract::<_, i16>(a.as_i16x8(), (imm8 & 7) as u32) as i32
1378+
simd_extract::<_, u16>(a.as_u16x8(), (imm8 & 7) as u32) as i32
13791379
}
13801380

13811381
/// Returns a new vector where the `imm8` element of `a` is replaced with `i`.
@@ -4132,7 +4132,7 @@ mod tests {
41324132
let a = _mm_setr_epi16(-1, 1, 2, 3, 4, 5, 6, 7);
41334133
let r1 = _mm_extract_epi16(a, 0);
41344134
let r2 = _mm_extract_epi16(a, 11);
4135-
assert_eq!(r1, -1);
4135+
assert_eq!(r1, 0xFFFF);
41364136
assert_eq!(r2, 3);
41374137
}
41384138

Diff for: crates/stdarch-verify/tests/x86-intel.rs

+11-4
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,8 @@ struct Data {
9595

9696
#[derive(Deserialize)]
9797
struct Intrinsic {
98-
rettype: String,
98+
#[serde(rename = "return")]
99+
return_: Return,
99100
name: String,
100101
#[serde(rename = "CPUID", default)]
101102
cpuid: Vec<String>,
@@ -111,6 +112,12 @@ struct Parameter {
111112
type_: String,
112113
}
113114

115+
#[derive(Deserialize)]
116+
struct Return {
117+
#[serde(rename = "type")]
118+
type_: String,
119+
}
120+
114121
#[derive(Deserialize, Debug)]
115122
struct Instruction {
116123
name: String,
@@ -503,12 +510,12 @@ fn matches(rust: &Function, intel: &Intrinsic) -> Result<(), String> {
503510

504511
// Make sure we've got the right return type.
505512
if let Some(t) = rust.ret {
506-
equate(t, &intel.rettype, rust.name, false)?;
507-
} else if intel.rettype != "" && intel.rettype != "void" {
513+
equate(t, &intel.return_.type_, rust.name, false)?;
514+
} else if intel.return_.type_ != "" && intel.return_.type_ != "void" {
508515
bail!(
509516
"{} returns `{}` with intel, void in rust",
510517
rust.name,
511-
intel.rettype
518+
intel.return_.type_
512519
)
513520
}
514521

0 commit comments

Comments
 (0)