rust/tests/ui/simd/issue-89193.rs

//@ run-pass

// Test that simd gather instructions on slice of usize don't cause crash
// See issue #89183 - https://github.com/rust-lang/rust/issues/89193

#![feature(repr_simd, core_intrinsics)]
#![allow(non_camel_case_types)]

use std::intrinsics::simd::simd_gather;

#[repr(simd)]
#[derive(Copy, Clone, PartialEq, Debug)]
struct x4<T>(pub [T; 4]);

fn main() {
    let x: [usize; 4] = [10, 11, 12, 13];
    let default = x4([0_usize, 1, 2, 3]);
    let all_set = u8::MAX as i8; // aka -1
    let mask = x4([all_set, all_set, all_set, all_set]);
    let expected = x4([10_usize, 11, 12, 13]);

    unsafe {
        let pointer = x.as_ptr();
        let pointers =
            x4([pointer.offset(0), pointer.offset(1), pointer.offset(2), pointer.offset(3)]);
        let result = simd_gather(default, pointers, mask);
        assert_eq!(result, expected);
    }

    // and again for isize
    let x: [isize; 4] = [10, 11, 12, 13];
    let default = x4([0_isize, 1, 2, 3]);
    let expected = x4([10_isize, 11, 12, 13]);

    unsafe {
        let pointer = x.as_ptr();
        let pointers =
            x4([pointer.offset(0), pointer.offset(1), pointer.offset(2), pointer.offset(3)]);
        let result = simd_gather(default, pointers, mask);
        assert_eq!(result, expected);
    }
}