ZLUDA/zluda_ml/src/lib.rs

35 lines
950 B
Rust
Raw Normal View History

2024-12-09 16:53:33 +00:00
mod r#impl;
macro_rules! unimplemented_fn {
($($abi:literal fn $fn_name:ident( $($arg_id:ident : $arg_type:ty),* ) -> $ret_type:ty;)*) => {
$(
#[no_mangle]
#[allow(improper_ctypes_definitions)]
pub extern $abi fn $fn_name ( $( $arg_id : $arg_type),* ) -> $ret_type {
r#impl::unimplemented()
}
)*
};
}
macro_rules! implemented_fn {
($($abi:literal fn $fn_name:ident( $($arg_id:ident : $arg_type:ty),* ) -> $ret_type:ty;)*) => {
$(
#[no_mangle]
#[allow(improper_ctypes_definitions)]
pub extern $abi fn $fn_name ( $( $arg_id : $arg_type),* ) -> $ret_type {
r#impl::$fn_name($($arg_id),*)
}
)*
};
}
cuda_base::nvml_function_declarations!(
unimplemented_fn,
implemented_fn <= [
2024-12-10 15:29:48 +00:00
nvmlErrorString,
nvmlInit_v2,
nvmlSystemGetDriverVersion
2024-12-09 16:53:33 +00:00
]
);