255 lines
6.5 KiB
Rust
255 lines
6.5 KiB
Rust
|
// assembly-output: ptx-linker
|
||
|
|
// compile-flags: --crate-type cdylib -C target-cpu=sm_86
|
||
|
|
// only-nvptx64
|
||
|
|
// ignore-nvptx64
|
||
|
|
|
||
|
|
// The following ABI tests are made with nvcc 11.6 does.
|
||
|
|
//
|
||
|
|
// The PTX ABI stability is tied to major versions of the PTX ISA
|
||
|
|
// These tests assume major version 7
|
||
|
|
//
|
||
|
|
//
|
||
|
|
// The following correspondence between types are assumed:
|
||
|
|
// u<N> - uint<N>_t
|
||
|
|
// i<N> - int<N>_t
|
||
|
|
// [T, N] - std::array<T, N>
|
||
|
|
// &T - T const*
|
||
|
|
// &mut T - T*
|
||
|
|
|
||
|
|
// CHECK: .version 7
|
||
|
|
|
||
|
|
#![feature(abi_ptx, lang_items, no_core)]
|
||
|
|
#![no_core]
|
||
|
|
|
||
|
|
#[lang = "sized"]
|
||
|
|
trait Sized {}
|
||
|
|
#[lang = "copy"]
|
||
|
|
trait Copy {}
|
||
|
|
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct SingleU8 {
|
||
|
|
f: u8,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct DoubleU8 {
|
||
|
|
f: u8,
|
||
|
|
g: u8,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct TripleU8 {
|
||
|
|
f: u8,
|
||
|
|
g: u8,
|
||
|
|
h: u8,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct TripleU16 {
|
||
|
|
f: u16,
|
||
|
|
g: u16,
|
||
|
|
h: u16,
|
||
|
|
}
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct TripleU32 {
|
||
|
|
f: u32,
|
||
|
|
g: u32,
|
||
|
|
h: u32,
|
||
|
|
}
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct TripleU64 {
|
||
|
|
f: u64,
|
||
|
|
g: u64,
|
||
|
|
h: u64,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct DoubleFloat {
|
||
|
|
f: f32,
|
||
|
|
g: f32,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct TripleFloat {
|
||
|
|
f: f32,
|
||
|
|
g: f32,
|
||
|
|
h: f32,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct TripleDouble {
|
||
|
|
f: f64,
|
||
|
|
g: f64,
|
||
|
|
h: f64,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct ManyIntegers {
|
||
|
|
f: u8,
|
||
|
|
g: u16,
|
||
|
|
h: u32,
|
||
|
|
i: u64,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[repr(C)]
|
||
|
|
pub struct ManyNumerics {
|
||
|
|
f: u8,
|
||
|
|
g: u16,
|
||
|
|
h: u32,
|
||
|
|
i: u64,
|
||
|
|
j: f32,
|
||
|
|
k: f64,
|
||
|
|
}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_u8_arg(
|
||
|
|
// CHECK: .param .u8 f_u8_arg_param_0
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_u8_arg(_a: u8) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_u16_arg(
|
||
|
|
// CHECK: .param .u16 f_u16_arg_param_0
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_u16_arg(_a: u16) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_u32_arg(
|
||
|
|
// CHECK: .param .u32 f_u32_arg_param_0
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_u32_arg(_a: u32) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_u64_arg(
|
||
|
|
// CHECK: .param .u64 f_u64_arg_param_0
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_u64_arg(_a: u64) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_u128_arg(
|
||
|
|
// CHECK: .param .align 16 .b8 f_u128_arg_param_0[16]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_u128_arg(_a: u128) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_i8_arg(
|
||
|
|
// CHECK: .param .u8 f_i8_arg_param_0
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_i8_arg(_a: i8) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_i16_arg(
|
||
|
|
// CHECK: .param .u16 f_i16_arg_param_0
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_i16_arg(_a: i16) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_i32_arg(
|
||
|
|
// CHECK: .param .u32 f_i32_arg_param_0
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_i32_arg(_a: i32) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_i64_arg(
|
||
|
|
// CHECK: .param .u64 f_i64_arg_param_0
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_i64_arg(_a: i64) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_i128_arg(
|
||
|
|
// CHECK: .param .align 16 .b8 f_i128_arg_param_0[16]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_i128_arg(_a: i128) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_f32_arg(
|
||
|
|
// CHECK: .param .f32 f_f32_arg_param_0
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_f32_arg(_a: f32) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_f64_arg(
|
||
|
|
// CHECK: .param .f64 f_f64_arg_param_0
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_f64_arg(_a: f64) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_single_u8_arg(
|
||
|
|
// CHECK: .param .align 1 .b8 f_single_u8_arg_param_0[1]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_single_u8_arg(_a: SingleU8) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_double_u8_arg(
|
||
|
|
// CHECK: .param .align 1 .b8 f_double_u8_arg_param_0[2]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_double_u8_arg(_a: DoubleU8) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_triple_u8_arg(
|
||
|
|
// CHECK: .param .align 1 .b8 f_triple_u8_arg_param_0[3]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_triple_u8_arg(_a: TripleU8) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_triple_u16_arg(
|
||
|
|
// CHECK: .param .align 2 .b8 f_triple_u16_arg_param_0[6]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_triple_u16_arg(_a: TripleU16) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_triple_u32_arg(
|
||
|
|
// CHECK: .param .align 4 .b8 f_triple_u32_arg_param_0[12]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_triple_u32_arg(_a: TripleU32) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_triple_u64_arg(
|
||
|
|
// CHECK: .param .align 8 .b8 f_triple_u64_arg_param_0[24]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_triple_u64_arg(_a: TripleU64) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_many_integers_arg(
|
||
|
|
// CHECK: .param .align 8 .b8 f_many_integers_arg_param_0[16]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_many_integers_arg(_a: ManyIntegers) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_double_float_arg(
|
||
|
|
// CHECK: .param .align 4 .b8 f_double_float_arg_param_0[8]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_double_float_arg(_a: DoubleFloat) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_triple_float_arg(
|
||
|
|
// CHECK: .param .align 4 .b8 f_triple_float_arg_param_0[12]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_triple_float_arg(_a: TripleFloat) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_triple_double_arg(
|
||
|
|
// CHECK: .param .align 8 .b8 f_triple_double_arg_param_0[24]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_triple_double_arg(_a: TripleDouble) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_many_numerics_arg(
|
||
|
|
// CHECK: .param .align 8 .b8 f_many_numerics_arg_param_0[32]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_many_numerics_arg(_a: ManyNumerics) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_byte_array_arg(
|
||
|
|
// CHECK: .param .align 1 .b8 f_byte_array_arg_param_0[5]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_byte_array_arg(_a: [u8; 5]) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_float_array_arg(
|
||
|
|
// CHECK: .param .align 4 .b8 f_float_array_arg_param_0[20]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_float_array_arg(_a: [f32; 5]) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_u128_array_arg(
|
||
|
|
// CHECK: .param .align 16 .b8 f_u128_array_arg_param_0[80]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_u128_array_arg(_a: [u128; 5]) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_u32_slice_arg(
|
||
|
|
// CHECK: .param .u64 f_u32_slice_arg_param_0
|
||
|
|
// CHECK: .param .u64 f_u32_slice_arg_param_1
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_u32_slice_arg(_a: &[u32]) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_tuple_u8_u8_arg(
|
||
|
|
// CHECK: .param .align 1 .b8 f_tuple_u8_u8_arg_param_0[2]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_tuple_u8_u8_arg(_a: (u8, u8)) {}
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_tuple_u32_u32_arg(
|
||
|
|
// CHECK: .param .align 4 .b8 f_tuple_u32_u32_arg_param_0[8]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_tuple_u32_u32_arg(_a: (u32, u32)) {}
|
||
|
|
|
||
|
|
|
||
|
|
// CHECK: .visible .entry f_tuple_u8_u8_u32_arg(
|
||
|
|
// CHECK: .param .align 4 .b8 f_tuple_u8_u8_u32_arg_param_0[8]
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "ptx-kernel" fn f_tuple_u8_u8_u32_arg(_a: (u8, u8, u32)) {}
|