1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

use core::ffi::c_void;

use half::f16;

pub trait KernelParam {
    fn to_launch_arg(&self) -> *mut c_void {
        self as *const _ as *mut _
    }
}

impl<T: KernelParam> KernelParam for *mut T {}
impl<T: KernelParam> KernelParam for *const T {}
impl KernelParam for f16 {}
impl KernelParam for f32 {}
impl KernelParam for f64 {}
impl KernelParam for u8 {}
impl KernelParam for u16 {}
impl KernelParam for u32 {}
impl KernelParam for u64 {}
impl KernelParam for i8 {}
impl KernelParam for i16 {}
impl KernelParam for i32 {}
impl KernelParam for i64 {}