forked from rust-lang/rust
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add test for asserting correct generation of ptx-kernel args
- Loading branch information
1 parent
352abba
commit 5bf5acc
Showing
1 changed file
with
254 additions
and
0 deletions.
There are no files selected for viewing
254 changes: 254 additions & 0 deletions
254
src/test/assembly/nvptx-kernel-abi/nvptx-kernel-args-abi-v7.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,254 @@ | ||
// 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)) {} |