about summary refs log tree commit diff
path: root/tests/ui/abi/simd-abi-checks-avx.rs
blob: 7432381d15b72cec1c0f3e3ef3a4377b3c03d4b2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
//@ only-x86_64
//@ build-fail
//@ compile-flags: -C target-feature=-avx

#![feature(portable_simd)]
#![feature(simd_ffi)]
#![allow(improper_ctypes_definitions)]

use std::arch::x86_64::*;

#[repr(transparent)]
struct Wrapper(__m256);

unsafe extern "C" fn w(_: Wrapper) {
    //~^ ERROR: requires the `avx` target feature, which is not enabled
    todo!()
}

unsafe extern "C" fn f(_: __m256) {
    //~^ ERROR: requires the `avx` target feature, which is not enabled
    todo!()
}

unsafe extern "C" fn g() -> __m256 {
    //~^ ERROR: requires the `avx` target feature, which is not enabled
    todo!()
}

#[target_feature(enable = "avx")]
unsafe extern "C" fn favx() -> __m256 {
    todo!()
}

// avx2 implies avx, so no error here.
#[target_feature(enable = "avx2")]
unsafe extern "C" fn gavx(_: __m256) {
    todo!()
}

// No error because of "Rust" ABI.
fn as_f64x8(d: __m512d) -> std::simd::f64x8 {
    unsafe { std::mem::transmute(d) }
}

unsafe fn test() {
    let arg = std::mem::transmute([0.0f64; 8]);
    as_f64x8(arg);
}

#[target_feature(enable = "avx")]
unsafe fn in_closure() -> impl FnOnce() -> __m256 {
    #[inline(always)] // this disables target-feature inheritance
    || g()
    //~^ ERROR requires the `avx` target feature, which is not enabled in the caller
}

fn main() {
    unsafe {
        f(g());
        //~^ ERROR requires the `avx` target feature, which is not enabled in the caller
        //~| ERROR requires the `avx` target feature, which is not enabled in the caller
    }

    unsafe {
        gavx(favx());
        //~^ ERROR requires the `avx` target feature, which is not enabled in the caller
        //~| ERROR requires the `avx` target feature, which is not enabled in the caller
    }

    unsafe {
        test();
    }

    unsafe {
        w(Wrapper(g()));
        //~^ ERROR requires the `avx` target feature, which is not enabled in the caller
        //~| ERROR requires the `avx` target feature, which is not enabled in the caller
    }

    unsafe {
        in_closure()();
    }

    unsafe {
        #[expect(improper_ctypes)]
        extern "C" {
            fn some_extern() -> __m256;
        }
        some_extern();
        //~^ ERROR requires the `avx` target feature, which is not enabled in the caller
    }
}

#[no_mangle]
#[target_feature(enable = "avx")]
fn some_extern() -> __m256 {
    todo!()
}