33add367e2
The `gpu-kernel` calling convention has several restrictions that were not enforced by the compiler until now. Add the following restrictions: 1. Cannot be async 2. Cannot be called 3. Cannot return values, return type must be `()` or `!` 4. Arguments should be primitives, i.e. passed by value. More complicated types can work when you know what you are doing, but it is rather unintuitive, one needs to know ABI/compiler internals. 5. Export name should be unmangled, either through `no_mangle` or `export_name`. Kernels are searched by name on the CPU side, having a mangled name makes it hard to find and probably almost always unintentional.
62 lines
2.1 KiB
Rust
62 lines
2.1 KiB
Rust
// Test argument and return type restrictions of the gpu-kernel ABI and
|
|
// check for warnings on mangled gpu-kernels.
|
|
|
|
//@ check-pass
|
|
//@ ignore-backends: gcc
|
|
//@ revisions: amdgpu nvptx
|
|
//@ add-minicore
|
|
//@ edition: 2024
|
|
//@[amdgpu] compile-flags: --target amdgcn-amd-amdhsa -Ctarget-cpu=gfx900
|
|
//@[amdgpu] needs-llvm-components: amdgpu
|
|
//@[nvptx] compile-flags: --target nvptx64-nvidia-cuda
|
|
//@[nvptx] needs-llvm-components: nvptx
|
|
|
|
#![feature(no_core, abi_gpu_kernel)]
|
|
#![no_core]
|
|
#![crate_type = "lib"]
|
|
|
|
extern crate minicore;
|
|
use minicore::*;
|
|
|
|
// Return types can be () or !
|
|
#[unsafe(no_mangle)]
|
|
extern "gpu-kernel" fn ret_empty() {}
|
|
#[unsafe(no_mangle)]
|
|
extern "gpu-kernel" fn ret_never() -> ! { loop {} }
|
|
|
|
// Arguments can be scalars or pointers
|
|
#[unsafe(no_mangle)]
|
|
extern "gpu-kernel" fn arg_i32(_: i32) { }
|
|
#[unsafe(no_mangle)]
|
|
extern "gpu-kernel" fn arg_ptr(_: *const i32) { }
|
|
#[unsafe(no_mangle)]
|
|
extern "gpu-kernel" fn arg_ptr_mut(_: *mut i32) { }
|
|
|
|
#[unsafe(no_mangle)]
|
|
extern "gpu-kernel" fn arg_zst(_: ()) { }
|
|
//~^ WARN passing type `()` to a function with "gpu-kernel" ABI may have unexpected behavior
|
|
//~^^ WARN `extern` fn uses type `()`, which is not FFI-safe
|
|
|
|
#[unsafe(no_mangle)]
|
|
extern "gpu-kernel" fn arg_ref(_: &i32) { }
|
|
//~^ WARN passing type `&i32` to a function with "gpu-kernel" ABI may have unexpected behavior
|
|
#[unsafe(no_mangle)]
|
|
extern "gpu-kernel" fn arg_ref_mut(_: &mut i32) { }
|
|
//~^ WARN passing type `&mut i32` to a function with "gpu-kernel" ABI may have unexpected behavior
|
|
|
|
struct S { a: i32, b: i32 }
|
|
#[unsafe(no_mangle)]
|
|
extern "gpu-kernel" fn arg_struct(_: S) { }
|
|
//~^ WARN passing type `S` to a function with "gpu-kernel" ABI may have unexpected behavior
|
|
//~^^ WARN `extern` fn uses type `S`, which is not FFI-safe
|
|
|
|
#[unsafe(no_mangle)]
|
|
extern "gpu-kernel" fn arg_tup(_: (i32, i32)) { }
|
|
//~^ WARN passing type `(i32, i32)` to a function with "gpu-kernel" ABI may have unexpected behavior
|
|
//~^^ WARN `extern` fn uses type `(i32, i32)`, which is not FFI-safe
|
|
|
|
#[unsafe(export_name = "kernel")]
|
|
pub extern "gpu-kernel" fn allowed_kernel_name() {}
|
|
|
|
pub extern "gpu-kernel" fn mangled_kernel() { }
|
|
//~^ WARN function with the "gpu-kernel" ABI has a mangled name
|