Applied-Software/ZLUDA
5
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-07-21 19:26:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

Change CUDA functions macro to use zluda_bindgen

Browse Source
This commit is contained in:
Andrzej Janik
2024-11-19 05:09:40 +00:00
parent aa6b61b414
commit 79362757ed
5 changed files with 20824 additions and 8989 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29035
cuda_base/src/cuda.rs
View File

File diff suppressed because it is too large Load Diff

659
cuda_base/src/lib.rs
View File

@ -1,165 +1,24 @@
extern crate proc_macro; extern crate proc_macro;
use std::collections::hash_map;
use std::iter;
use proc_macro::TokenStream; use proc_macro::TokenStream;
use proc_macro2::Span; use quote::{quote, ToTokens};
use quote::{format_ident, quote, ToTokens}; use rustc_hash::FxHashMap;
use rustc_hash::{FxHashMap, FxHashSet}; use std::iter;
use syn::parse::{Parse, ParseStream}; use syn::parse::{Parse, ParseStream};
use syn::punctuated::Punctuated; use syn::punctuated::Punctuated;
use syn::visit_mut::VisitMut; use syn::visit_mut::VisitMut;
use syn::{ use syn::{
bracketed, parse_macro_input, parse_quote, Abi, Fields, File, FnArg, ForeignItem, bracketed, parse_macro_input, File, ForeignItem, ForeignItemFn, Ident, Item, Path, Signature,
ForeignItemFn, Ident, Item, ItemConst, ItemForeignMod, LitStr, PatType, Path, PathArguments, Token,
PathSegment, ReturnType, Signature, Token, Type, TypeArray, TypePath, TypePtr, UseTree,
}; };
const CUDA_RS: &'static str = include_str! {"cuda.rs"}; const CUDA_RS: &'static str = include_str! {"cuda.rs"};
// This macro copies cuda.rs as-is with some changes:
// * All function declarations are filtered out
// * CUdeviceptr_v2 is redefined from `unsigned long long` to `*void`
// * `extern "C"` gets replaced by `extern "system"`
// * CUuuid_st is redefined to use uchar instead of char
#[proc_macro]
pub fn cuda_type_declarations(_: TokenStream) -> TokenStream {
let mut cuda_module = syn::parse_str::<File>(CUDA_RS).unwrap();
let mut curesult_constants = Vec::new();
cuda_module.items = cuda_module
.items
.into_iter()
.filter_map(|item| match item {
Item::ForeignMod(_) => None,
Item::Struct(mut struct_) => {
let ident_string = struct_.ident.to_string();
match &*ident_string {
"CUdeviceptr_v2" => match &mut struct_.fields {
Fields::Unnamed(ref mut fields) => {
fields.unnamed[0].ty =
absolute_path_to_mut_ptr(&["std", "os", "raw", "c_void"])
}
_ => unreachable!(),
},
"CUuuid_st" => match &mut struct_.fields {
Fields::Named(ref mut fields) => match fields.named[0].ty {
Type::Array(TypeArray { ref mut elem, .. }) => {
*elem = Box::new(Type::Path(TypePath {
qself: None,
path: segments_to_path(&["std", "os", "raw", "c_uchar"]),
}))
}
_ => unreachable!(),
},
_ => panic!(),
},
_ => {}
}
Some(Item::Struct(struct_))
}
Item::Const(const_) => {
let name = const_.ident.to_string();
if name.starts_with("cudaError_enum_CUDA_") {
curesult_constants.push(const_);
}
None
}
Item::Use(use_) => {
if let UseTree::Path(ref path) = use_.tree {
if let UseTree::Rename(ref rename) = &*path.tree {
if rename.rename == "CUresult" {
return None;
}
}
}
Some(Item::Use(use_))
}
i => Some(i),
})
.collect::<Vec<_>>();
append_curesult(curesult_constants, &mut cuda_module.items);
syn::visit_mut::visit_file_mut(&mut FixAbi, &mut cuda_module);
cuda_module.into_token_stream().into()
}
fn append_curesult(curesult_constants: Vec<syn::ItemConst>, items: &mut Vec<Item>) {
let curesult_constants = curesult_constants.iter().map(|const_| {
let ident = const_.ident.to_string();
let expr = &const_.expr;
if ident.ends_with("CUDA_SUCCESS") {
quote! {
const SUCCESS: CUresult = CUresult::Ok(());
}
} else {
let prefix = "cudaError_enum_CUDA_";
let ident = format_ident!("{}", ident[prefix.len()..]);
quote! {
const #ident: CUresult = CUresult::Err(unsafe { ::core::num::NonZeroU32::new_unchecked(#expr) });
}
}
});
items.push(parse_quote! {
pub trait CUresultConsts {
#(#curesult_constants)*
}
});
items.push(parse_quote! {
impl CUresultConsts for CUresult {}
});
items.push(parse_quote! {
#[must_use]
pub type CUresult = ::core::result::Result<(), ::core::num::NonZeroU32>;
});
items.push(parse_quote! {
const _: fn() = || {
let _ = std::mem::transmute::<CUresult, u32>;
};
});
}
fn segments_to_path(path: &[&'static str]) -> Path {
let mut segments = Punctuated::new();
for ident in path {
let ident = PathSegment {
ident: Ident::new(ident, Span::call_site()),
arguments: PathArguments::None,
};
segments.push(ident);
}
Path {
leading_colon: Some(Token![::](Span::call_site())),
segments,
}
}
fn absolute_path_to_mut_ptr(path: &[&'static str]) -> Type {
Type::Ptr(TypePtr {
star_token: Token![*](Span::call_site()),
const_token: None,
mutability: Some(Token![mut](Span::call_site())),
elem: Box::new(Type::Path(TypePath {
qself: None,
path: segments_to_path(path),
})),
})
}
struct FixAbi;
impl VisitMut for FixAbi {
fn visit_abi_mut(&mut self, i: &mut Abi) {
if let Some(ref mut name) = i.name {
*name = LitStr::new("system", Span::call_site());
}
}
}
// This macro accepts following arguments: // This macro accepts following arguments:
// * `type_path`: path to the module with type definitions (in the module tree)
// * `normal_macro`: ident for a normal macro // * `normal_macro`: ident for a normal macro
// * `override_macro`: ident for an override macro // * zero or more:
// * `override_fns`: list of override functions // * `override_macro`: ident for an override macro
// * `override_fns`: list of override functions
// Then macro goes through every function in rust.rs, and for every fn `foo`: // Then macro goes through every function in rust.rs, and for every fn `foo`:
// * if `foo` is contained in `override_fns` then pass it into `override_macro` // * if `foo` is contained in `override_fns` then pass it into `override_macro`
// * if `foo` is not contained in `override_fns` pass it to `normal_macro` // * if `foo` is not contained in `override_fns` pass it to `normal_macro`
@ -172,430 +31,120 @@ impl VisitMut for FixAbi {
#[proc_macro] #[proc_macro]
pub fn cuda_function_declarations(tokens: TokenStream) -> TokenStream { pub fn cuda_function_declarations(tokens: TokenStream) -> TokenStream {
let input = parse_macro_input!(tokens as FnDeclInput); let input = parse_macro_input!(tokens as FnDeclInput);
let cuda_module = syn::parse_str::<File>(CUDA_RS).unwrap(); let mut choose_macro = ChooseMacro::new(input);
let override_fns = input let mut cuda_module = syn::parse_str::<File>(CUDA_RS).unwrap();
.override_fns syn::visit_mut::visit_file_mut(&mut FixFnSignatures, &mut cuda_module);
.iter() let extern_ = if let Item::ForeignMod(extern_) = cuda_module.items.pop().unwrap() {
.map(ToString::to_string) extern_
.collect::<FxHashSet<_>>(); } else {
let (normal_macro_args, override_macro_args): (Vec<_>, Vec<_>) = cuda_module unreachable!()
.items };
.into_iter() let abi = extern_.abi.name;
.filter_map(|item| match item { for mut item in extern_.items {
Item::ForeignMod(ItemForeignMod { mut items, .. }) => match items.pop().unwrap() { if let ForeignItem::Fn(ForeignItemFn {
ForeignItem::Fn(ForeignItemFn { sig: Signature { ref ident, .. },
sig: ref mut attrs,
Signature { ..
ident, }) = item
inputs, {
output, *attrs = Vec::new();
.. choose_macro.add(ident, quote! { #abi #item });
}, } else {
.. unreachable!()
}) => { }
let use_normal_macro = !override_fns.contains(&ident.to_string());
let inputs = inputs
.into_iter()
.map(|fn_arg| match fn_arg {
FnArg::Typed(mut pat_type) => {
pat_type.ty =
prepend_cuda_path_to_type(&input.type_path, pat_type.ty);
FnArg::Typed(pat_type)
}
_ => unreachable!(),
})
.collect::<Punctuated<_, Token![,]>>();
let output = match output {
ReturnType::Type(_, type_) => type_,
ReturnType::Default => unreachable!(),
};
let type_path = input.type_path.clone();
Some((
quote! {
"system" fn #ident(#inputs) -> #type_path :: #output
},
use_normal_macro,
))
}
_ => unreachable!(),
},
_ => None,
})
.partition(|(_, use_normal_macro)| *use_normal_macro);
let mut result = proc_macro2::TokenStream::new();
if !normal_macro_args.is_empty() {
let punctuated_normal_macro_args = to_punctuated::<Token![;]>(normal_macro_args);
let macro_ = &input.normal_macro;
result.extend(iter::once(quote! {
#macro_ ! (#punctuated_normal_macro_args);
}));
} }
if !override_macro_args.is_empty() { let mut result = proc_macro2::TokenStream::new();
let punctuated_override_macro_args = to_punctuated::<Token![;]>(override_macro_args); for (path, items) in
let macro_ = &input.override_macro; iter::once(choose_macro.default).chain(choose_macro.override_sets.into_iter())
result.extend(iter::once(quote! { {
#macro_ ! (#punctuated_override_macro_args); if items.is_empty() {
})); continue;
}
quote! {
#path ! { #(#items)* }
}
.to_tokens(&mut result);
} }
result.into() result.into()
} }
fn to_punctuated<P: ToTokens + Default>(
elms: Vec<(proc_macro2::TokenStream, bool)>,
) -> proc_macro2::TokenStream {
let mut collection = Punctuated::<proc_macro2::TokenStream, P>::new();
collection.extend(elms.into_iter().map(|(token_stream, _)| token_stream));
collection.into_token_stream()
}
fn prepend_cuda_path_to_type(base_path: &Path, type_: Box<Type>) -> Box<Type> {
match *type_ {
Type::Path(mut type_path) => {
type_path.path = prepend_cuda_path_to_path(base_path, type_path.path);
Box::new(Type::Path(type_path))
}
Type::Ptr(mut type_ptr) => {
type_ptr.elem = prepend_cuda_path_to_type(base_path, type_ptr.elem);
Box::new(Type::Ptr(type_ptr))
}
_ => unreachable!(),
}
}
fn prepend_cuda_path_to_path(base_path: &Path, path: Path) -> Path {
if path.leading_colon.is_some() {
return path;
}
if path.segments.len() == 1 {
let ident = path.segments[0].ident.to_string();
if ident.starts_with("CU")
|| ident.starts_with("cu")
|| ident.starts_with("GL")
|| ident.starts_with("EGL")
|| ident.starts_with("Vdp")
|| ident == "HGPUNV"
{
let mut base_path = base_path.clone();
base_path.segments.extend(path.segments);
return base_path;
}
}
path
}
struct FnDeclInput { struct FnDeclInput {
type_path: Path,
normal_macro: Path, normal_macro: Path,
override_macro: Path, overrides: Punctuated<OverrideMacro, Token![,]>,
override_fns: Punctuated<Ident, Token![,]>,
} }
impl Parse for FnDeclInput { impl Parse for FnDeclInput {
fn parse(input: ParseStream) -> syn::Result<Self> { fn parse(input: ParseStream) -> syn::Result<Self> {
let type_path = input.parse::<Path>()?;
input.parse::<Token![,]>()?;
let normal_macro = input.parse::<Path>()?; let normal_macro = input.parse::<Path>()?;
input.parse::<Token![,]>()?; let overrides = if input.is_empty() {
let override_macro = input.parse::<Path>()?; Punctuated::new()
input.parse::<Token![,]>()?; } else {
let override_fns_content; input.parse::<Token![,]>()?;
bracketed!(override_fns_content in input); input.parse_terminated(OverrideMacro::parse, Token![,])?
let override_fns = override_fns_content.parse_terminated(Ident::parse, Token![,])?; };
Ok(Self { Ok(Self {
type_path,
normal_macro, normal_macro,
override_macro, overrides,
override_fns,
}) })
} }
} }
struct OverrideMacro {
// This trait accepts following parameters: macro_: Path,
// * `type_path`: path to the module with type definitions (in the module tree) functions: Punctuated<Ident, Token![,]>,
// * `trait_`: name of the trait to be derived
// * `ignore_types`: bracketed list of types to ignore
// * `ignore_fns`: bracketed list of fns to ignore
#[proc_macro]
pub fn cuda_derive_display_trait(tokens: TokenStream) -> TokenStream {
let input = parse_macro_input!(tokens as DeriveDisplayInput);
let cuda_module = syn::parse_str::<File>(CUDA_RS).unwrap();
let mut derive_state = DeriveDisplayState::new(input);
let mut main_body: proc_macro2::TokenStream = cuda_module
.items
.into_iter()
.filter_map(|i| cuda_derive_display_trait_for_item(&mut derive_state, i))
.collect::<proc_macro2::TokenStream>();
main_body.extend(curesult_display_trait(&derive_state));
main_body.into()
} }
fn curesult_display_trait(derive_state: &DeriveDisplayState) -> proc_macro2::TokenStream { impl Parse for OverrideMacro {
let path_prefix = &derive_state.type_path;
let trait_ = &derive_state.trait_;
let errors = derive_state.result_variants.iter().filter_map(|const_| {
let prefix = "cudaError_enum_";
let text = &const_.ident.to_string()[prefix.len()..];
if text == "CUDA_SUCCESS" {
return None;
}
let expr = &const_.expr;
Some(quote! {
#expr => writer.write_all(#text.as_bytes()),
})
});
quote! {
impl #trait_ for #path_prefix :: CUresult {
fn write(&self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized)) -> std::io::Result<()> {
match self {
Ok(()) => writer.write_all(b"CUDA_SUCCESS"),
Err(err) => {
match err.get() {
#(#errors)*
err => write!(writer, "{}", err)
}
}
}
}
}
}
}
fn cuda_derive_display_trait_for_item(
state: &mut DeriveDisplayState,
item: Item,
) -> Option<proc_macro2::TokenStream> {
let path_prefix = &state.type_path;
let path_prefix_iter = iter::repeat(&path_prefix);
let trait_ = &state.trait_;
let trait_iter = iter::repeat(&state.trait_);
match item {
Item::Const(const_) => {
if const_.ty.to_token_stream().to_string() == "cudaError_enum" {
state.result_variants.push(const_);
}
None
}
Item::ForeignMod(ItemForeignMod { mut items, .. }) => match items.pop().unwrap() {
ForeignItem::Fn(ForeignItemFn {
sig: Signature { ident, inputs, .. },
..
}) => {
if state.ignore_fns.contains(&ident) {
return None;
}
let inputs = inputs
.into_iter()
.map(|fn_arg| match fn_arg {
FnArg::Typed(mut pat_type) => {
pat_type.ty = prepend_cuda_path_to_type(path_prefix, pat_type.ty);
FnArg::Typed(pat_type)
}
_ => unreachable!(),
})
.collect::<Vec<_>>();
let inputs_iter = inputs.iter();
let mut arg_name_iter = inputs.iter().map(|fn_arg| match fn_arg {
FnArg::Typed(PatType { pat, .. }) => pat,
_ => unreachable!(),
});
let fn_name = format_ident!("write_{}", ident);
let original_fn_name = ident.to_string();
Some(match arg_name_iter.next() {
Some(first_arg_name) => quote! {
pub fn #fn_name(writer: &mut (impl std::io::Write + ?Sized), #(#inputs_iter,)*) -> std::io::Result<()> {
writer.write_all(concat!("(", stringify!(#first_arg_name), ": ").as_bytes())?;
let mut arg_idx = 0usize;
CudaDisplay::write(&#first_arg_name, #original_fn_name, arg_idx, writer)?;
#(
writer.write_all(b", ")?;
writer.write_all(concat!(stringify!(#arg_name_iter), ": ").as_bytes())?;
CudaDisplay::write(&#arg_name_iter, #original_fn_name, arg_idx, writer)?;
arg_idx += 1;
)*
writer.write_all(b")")
}
},
None => quote! {
pub fn #fn_name(writer: &mut (impl std::io::Write + ?Sized)) -> std::io::Result<()> {
writer.write_all(b"()")
}
},
})
}
_ => unreachable!(),
},
Item::Impl(mut item_impl) => {
let enum_ = match *(item_impl.self_ty) {
Type::Path(mut path) => path.path.segments.pop().unwrap().into_value().ident,
_ => unreachable!(),
};
let variant_ = match item_impl.items.pop().unwrap() {
syn::ImplItem::Const(item_const) => item_const.ident,
_ => unreachable!(),
};
state.record_enum_variant(enum_, variant_);
None
}
Item::Struct(item_struct) => {
if state.ignore_types.contains(&item_struct.ident) {
return None;
}
if state.enums.contains_key(&item_struct.ident) {
let enum_ = &item_struct.ident;
let enum_iter = iter::repeat(&item_struct.ident);
let variants = state.enums.get(&item_struct.ident).unwrap().iter();
Some(quote! {
impl #trait_ for #path_prefix :: #enum_ {
fn write(&self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized)) -> std::io::Result<()> {
match self {
#(& #path_prefix_iter :: #enum_iter :: #variants => writer.write_all(stringify!(#variants).as_bytes()),)*
_ => write!(writer, "{}", self.0)
}
}
}
})
} else {
let struct_ = &item_struct.ident;
let (first_field, rest_of_fields) = match item_struct.fields {
Fields::Named(fields) => {
let mut all_idents = fields.named.into_iter().filter_map(|f| {
let f_ident = f.ident.unwrap();
let name = f_ident.to_string();
if name.starts_with("reserved") || name == "_unused" {
None
} else {
Some(f_ident)
}
});
let first = match all_idents.next() {
Some(f) => f,
None => return None,
};
(first, all_idents)
}
_ => return None,
};
Some(quote! {
impl #trait_ for #path_prefix :: #struct_ {
fn write(&self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized)) -> std::io::Result<()> {
writer.write_all(concat!("{ ", stringify!(#first_field), ": ").as_bytes())?;
#trait_::write(&self.#first_field, "", 0, writer)?;
#(
writer.write_all(concat!(", ", stringify!(#rest_of_fields), ": ").as_bytes())?;
#trait_iter::write(&self.#rest_of_fields, "", 0, writer)?;
)*
writer.write_all(b" }")
}
}
})
}
}
Item::Type(item_type) => {
if state.ignore_types.contains(&item_type.ident) {
return None;
};
match *(item_type.ty) {
Type::Ptr(_) => {
let type_ = item_type.ident;
Some(quote! {
impl #trait_ for #path_prefix :: #type_ {
fn write(&self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized)) -> std::io::Result<()> {
write!(writer, "{:p}", *self)
}
}
})
}
Type::Path(type_path) => {
if type_path.path.leading_colon.is_some() {
let option_seg = type_path.path.segments.last().unwrap();
if option_seg.ident == "Option" {
match &option_seg.arguments {
PathArguments::AngleBracketed(generic) => match generic.args[0] {
syn::GenericArgument::Type(Type::BareFn(_)) => {
let type_ = &item_type.ident;
return Some(quote! {
impl #trait_ for #path_prefix :: #type_ {
fn write(&self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized)) -> std::io::Result<()> {
write!(writer, "{:p}", unsafe { std::mem::transmute::<#path_prefix :: #type_, *mut ::std::ffi::c_void>(*self) })
}
}
});
}
_ => unreachable!(),
},
_ => unreachable!(),
}
}
}
None
}
_ => unreachable!(),
}
}
Item::Union(_) => None,
Item::Use(_) => None,
_ => unreachable!(),
}
}
struct DeriveDisplayState {
type_path: Path,
trait_: Path,
ignore_types: FxHashSet<Ident>,
ignore_fns: FxHashSet<Ident>,
enums: FxHashMap<Ident, Vec<Ident>>,
result_variants: Vec<ItemConst>,
}
impl DeriveDisplayState {
fn new(input: DeriveDisplayInput) -> Self {
DeriveDisplayState {
type_path: input.type_path,
trait_: input.trait_,
ignore_types: input.ignore_types.into_iter().collect(),
ignore_fns: input.ignore_fns.into_iter().collect(),
enums: Default::default(),
result_variants: Vec::new(),
}
}
fn record_enum_variant(&mut self, enum_: Ident, variant: Ident) {
match self.enums.entry(enum_) {
hash_map::Entry::Occupied(mut entry) => {
entry.get_mut().push(variant);
}
hash_map::Entry::Vacant(entry) => {
entry.insert(vec![variant]);
}
}
}
}
struct DeriveDisplayInput {
type_path: Path,
trait_: Path,
ignore_types: Punctuated<Ident, Token![,]>,
ignore_fns: Punctuated<Ident, Token![,]>,
}
impl Parse for DeriveDisplayInput {
fn parse(input: ParseStream) -> syn::Result<Self> { fn parse(input: ParseStream) -> syn::Result<Self> {
let type_path = input.parse::<Path>()?; let macro_ = input.parse::<Path>()?;
input.parse::<Token![,]>()?; input.parse::<Token![<=]>()?;
let trait_ = input.parse::<Path>()?; let functions_content;
input.parse::<Token![,]>()?; bracketed!(functions_content in input);
let ignore_types_buffer; let functions = functions_content.parse_terminated(Ident::parse, Token![,])?;
bracketed!(ignore_types_buffer in input); Ok(Self { macro_, functions })
let ignore_types = ignore_types_buffer.parse_terminated(Ident::parse, Token![,])?; }
input.parse::<Token![,]>()?; }
let ignore_fns_buffer;
bracketed!(ignore_fns_buffer in input); struct ChooseMacro {
let ignore_fns = ignore_fns_buffer.parse_terminated(Ident::parse, Token![,])?; default: (Path, Vec<proc_macro2::TokenStream>),
Ok(Self { override_lookup: FxHashMap<Ident, Path>,
type_path, override_sets: FxHashMap<Path, Vec<proc_macro2::TokenStream>>,
trait_, }
ignore_types,
ignore_fns, impl ChooseMacro {
}) fn new(input: FnDeclInput) -> Self {
let mut override_lookup = FxHashMap::default();
let mut override_sets = FxHashMap::default();
for OverrideMacro { macro_, functions } in input.overrides {
for ident in functions {
override_lookup.insert(ident, macro_.clone());
override_sets.insert(macro_.clone(), Vec::new());
}
}
Self {
default: (input.normal_macro, Vec::new()),
override_lookup,
override_sets,
}
}
fn add(&mut self, ident: &Ident, tokens: proc_macro2::TokenStream) {
match self.override_lookup.get(ident) {
Some(override_macro) => {
self.override_sets
.get_mut(override_macro)
.unwrap()
.push(tokens);
}
None => self.default.1.push(tokens),
}
}
}
// For some reason prettyplease will append trailing comma *only*
// if there are two or more arguments
struct FixFnSignatures;
impl VisitMut for FixFnSignatures {
fn visit_signature_mut(&mut self, s: &mut syn::Signature) {
s.inputs.pop_punct();
} }
} }

107
zluda_bindgen/src/main.rs
View File

@ -5,7 +5,7 @@ use std::{collections::hash_map, fs::File, io::Write, iter, path::PathBuf, str::
use syn::{ use syn::{
parse_quote, punctuated::Punctuated, visit_mut::VisitMut, Abi, Fields, FnArg, ForeignItem, parse_quote, punctuated::Punctuated, visit_mut::VisitMut, Abi, Fields, FnArg, ForeignItem,
ForeignItemFn, Ident, Item, ItemConst, ItemForeignMod, ItemUse, LitStr, Path, PathArguments, ForeignItemFn, Ident, Item, ItemConst, ItemForeignMod, ItemUse, LitStr, Path, PathArguments,
Signature, Type, UseTree, Signature, Type, TypePath, UseTree,
}; };
fn main() { fn main() {
@ -32,6 +32,11 @@ fn main() {
.unwrap() .unwrap()
.to_string(); .to_string();
let module: syn::File = syn::parse_str(&cuda_header).unwrap(); let module: syn::File = syn::parse_str(&cuda_header).unwrap();
generate_functions(
&crate_root,
&["..", "cuda_base", "src", "cuda.rs"],
&module,
);
generate_types(&crate_root, &["..", "cuda_types", "src", "lib.rs"], &module); generate_types(&crate_root, &["..", "cuda_types", "src", "lib.rs"], &module);
generate_display( generate_display(
&crate_root, &crate_root,
@ -41,6 +46,27 @@ fn main() {
) )
} }
fn generate_functions(output: &PathBuf, path: &[&str], module: &syn::File) {
let fns_ = module.items.iter().filter_map(|item| match item {
Item::ForeignMod(extern_) => match &*extern_.items {
[ForeignItem::Fn(fn_)] => Some(fn_),
_ => unreachable!(),
},
_ => None,
});
let mut module: syn::File = parse_quote! {
extern "system" {
#(#fns_)*
}
};
syn::visit_mut::visit_file_mut(&mut PrependCudaPath, &mut module);
syn::visit_mut::visit_file_mut(&mut RemoveVisibility, &mut module);
syn::visit_mut::visit_file_mut(&mut ExplicitReturnType, &mut module);
let mut output = output.clone();
output.extend(path);
write_rust_to_file(output, &prettyplease::unparse(&module))
}
fn generate_types(output: &PathBuf, path: &[&str], module: &syn::File) { fn generate_types(output: &PathBuf, path: &[&str], module: &syn::File) {
let mut module = module.clone(); let mut module = module.clone();
let mut converter = ConvertIntoRustResult { let mut converter = ConvertIntoRustResult {
@ -181,6 +207,39 @@ impl VisitMut for FixAbi {
} }
} }
struct PrependCudaPath;
impl VisitMut for PrependCudaPath {
fn visit_type_path_mut(&mut self, type_: &mut TypePath) {
if type_.path.segments.len() == 1 {
match &*type_.path.segments[0].ident.to_string() {
"usize" | "f64" | "f32" => {}
_ => {
*type_ = parse_quote! { cuda_types :: #type_ };
}
}
}
}
}
struct RemoveVisibility;
impl VisitMut for RemoveVisibility {
fn visit_visibility_mut(&mut self, i: &mut syn::Visibility) {
*i = syn::Visibility::Inherited;
}
}
struct ExplicitReturnType;
impl VisitMut for ExplicitReturnType {
fn visit_return_type_mut(&mut self, i: &mut syn::ReturnType) {
if let syn::ReturnType::Default = i {
*i = parse_quote! { -> {} };
}
}
}
fn generate_display( fn generate_display(
output: &PathBuf, output: &PathBuf,
path: &[&str], path: &[&str],
@ -320,13 +379,10 @@ fn cuda_derive_display_trait_for_item<'a>(
} }
let inputs = inputs let inputs = inputs
.iter() .iter()
.map(|fn_arg| match fn_arg { .map(|fn_arg| {
FnArg::Typed(ref pat_type) => { let mut fn_arg = fn_arg.clone();
let mut pat_type = pat_type.clone(); syn::visit_mut::visit_fn_arg_mut(&mut PrependCudaPath, &mut fn_arg);
pat_type.ty = prepend_cuda_path_to_type(&path_prefix, pat_type.ty); fn_arg
FnArg::Typed(pat_type)
}
_ => unreachable!(),
}) })
.collect::<Vec<_>>(); .collect::<Vec<_>>();
let inputs_iter = inputs.iter(); let inputs_iter = inputs.iter();
@ -500,41 +556,6 @@ fn fn_arg_name(fn_arg: &FnArg) -> &Box<syn::Pat> {
name name
} }
fn prepend_cuda_path_to_type(base_path: &Path, type_: Box<Type>) -> Box<Type> {
match *type_ {
Type::Path(mut type_path) => {
type_path.path = prepend_cuda_path_to_path(base_path, type_path.path);
Box::new(Type::Path(type_path))
}
Type::Ptr(mut type_ptr) => {
type_ptr.elem = prepend_cuda_path_to_type(base_path, type_ptr.elem);
Box::new(Type::Ptr(type_ptr))
}
_ => unreachable!(),
}
}
fn prepend_cuda_path_to_path(base_path: &Path, path: Path) -> Path {
if path.leading_colon.is_some() {
return path;
}
if path.segments.len() == 1 {
let ident = path.segments[0].ident.to_string();
if ident.starts_with("CU")
|| ident.starts_with("cu")
|| ident.starts_with("GL")
|| ident.starts_with("EGL")
|| ident.starts_with("Vdp")
|| ident == "HGPUNV"
{
let mut base_path = base_path.clone();
base_path.segments.extend(path.segments);
return base_path;
}
}
path
}
fn curesult_display_trait(derive_state: &DeriveDisplayState) -> syn::Item { fn curesult_display_trait(derive_state: &DeriveDisplayState) -> syn::Item {
let errors = derive_state.result_variants.iter().filter_map(|const_| { let errors = derive_state.result_variants.iter().filter_map(|const_| {
let prefix = "cudaError_enum_"; let prefix = "cudaError_enum_";

8
zluda_dump/src/lib.rs
View File

@ -9,7 +9,7 @@ extern crate lazy_static;
extern crate cuda_types; extern crate cuda_types;
macro_rules! extern_redirect { macro_rules! extern_redirect {
($($abi:literal fn $fn_name:ident( $($arg_id:ident : $arg_type:ty),* ) -> $ret_type:path);*) => { ($($abi:literal fn $fn_name:ident( $($arg_id:ident : $arg_type:ty),* ) -> $ret_type:path;)*) => {
$( $(
#[no_mangle] #[no_mangle]
#[allow(improper_ctypes_definitions)] #[allow(improper_ctypes_definitions)]
@ -30,7 +30,7 @@ macro_rules! extern_redirect {
} }
macro_rules! extern_redirect_with_post { macro_rules! extern_redirect_with_post {
($($abi:literal fn $fn_name:ident( $($arg_id:ident : $arg_type:ty),* ) -> $ret_type:path);*) => { ($($abi:literal fn $fn_name:ident( $($arg_id:ident : $arg_type:ty),* ) -> $ret_type:path;)*) => {
$( $(
#[no_mangle] #[no_mangle]
#[allow(improper_ctypes_definitions)] #[allow(improper_ctypes_definitions)]
@ -57,10 +57,8 @@ macro_rules! extern_redirect_with_post {
use cuda_base::cuda_function_declarations; use cuda_base::cuda_function_declarations;
cuda_function_declarations!( cuda_function_declarations!(
cuda_types,
extern_redirect, extern_redirect,
extern_redirect_with_post, extern_redirect_with_post <= [
[
cuModuleLoad, cuModuleLoad,
cuModuleLoadData, cuModuleLoadData,
cuModuleLoadDataEx, cuModuleLoadDataEx,

4
zluda_dump/src/side_by_side.rs
View File

@ -56,7 +56,7 @@ impl CudaDynamicFns {
} }
macro_rules! emit_cuda_fn_table { macro_rules! emit_cuda_fn_table {
($($abi:literal fn $fn_name:ident( $($arg_id:ident : $arg_type:ty),* ) -> $ret_type:path);*) => { ($($abi:literal fn $fn_name:ident( $($arg_id:ident : $arg_type:ty),* ) -> $ret_type:path;)*) => {
#[derive(Default)] #[derive(Default)]
#[allow(improper_ctypes)] #[allow(improper_ctypes)]
#[allow(improper_ctypes_definitions)] #[allow(improper_ctypes_definitions)]
@ -76,4 +76,4 @@ macro_rules! emit_cuda_fn_table {
}; };
} }
cuda_function_declarations!(cuda_types, emit_cuda_fn_table, emit_cuda_fn_table, []); cuda_function_declarations!(emit_cuda_fn_table);