hanzo-rocm 0.5.2

Rust bindings for AMD ROCm libraries
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

/// impl helper macro for rocblas functions
#[macro_export]
macro_rules! impl_rocblas_func {
    ($trait_name:ident, $fn_type:ident, {$( $t:ty => $func:path ),* $(,)?}) => {
        $(
            impl $trait_name for $t {
                fn func() -> $fn_type<Self> {
                    $func
                }
            }
        )*
    };
}
#[macro_export]
macro_rules! impl_rocblas_func_inner {
    ($func:expr, $($arg:expr),+ $(,)?) => {{
        let status = unsafe { $func($($arg),+) };
        if status != ffi::rocblas_status__rocblas_status_success {
            return Err(Error::new(status));
        }
        Ok(())
    }};
}
#[macro_export]
macro_rules! impl_rocblas_traits {
    (
        $trait_name:ident,
        $fn_type:ident,
        $ffi_map:tt,
        $method_name:ident,
        ($($arg:ident : $arg_ty:ty),+ $(,)?),
        ($($fn_arg:ty),+ $(,)?),
        ($($call_arg:expr),+ $(,)?)
    ) => {
        type $fn_type<T> = unsafe extern "C" fn($($fn_arg),+) -> ffi::rocblas_status;

        pub trait $trait_name {
            fn func() -> $fn_type<Self>;

            unsafe fn $method_name(
                $($arg: $arg_ty),+
            ) -> Result<()> {
                impl_rocblas_func_inner!(
                    Self::func(),
                    $($call_arg),+
                )
            }
        }

        impl_rocblas_func!($trait_name, $fn_type, $ffi_map);
    };
}