dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix module derive with generics (#2127) 

* Remove unnecessary ModuleDisplayDefault generic bound + duplicate ModuleDisplay

* Remove erroneous bound for autodiff module generic
This commit is contained in:
Guillaume Lagrange 2024-08-08 16:24:51 -04:00 committed by GitHub
parent bb4a605ca6
commit 723c9d1a2e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
crates/burn-derive/src/module/codegen.rs
View File

@ -199,19 +199,12 @@ impl GenericsParser {
}
);
module.add_predicate(
parse_quote! {
#ident: burn::module::ModuleDisplayDefault
}
);
module.add_predicate(
parse_quote! {
#ident: burn::module::ModuleDisplay
}
);
module_autodiff.add_predicate(
parse_quote! {
#ident: burn::module::AutodiffModule<B>
@ -230,25 +223,7 @@ impl GenericsParser {
}
);
module_autodiff.add_predicate(
parse_quote! {
<#ident as burn::module::AutodiffModule<B>>::InnerModule: burn::module::ModuleDisplay
}
);
generics_names_except_backend.extend(quote! { <#ident as burn::module::AutodiffModule<B>>::InnerModule, });
module_autodiff.add_predicate(
parse_quote! {
#ident: burn::module::Module<B::InnerBackend>
}
);
module_autodiff.add_predicate(
parse_quote! {
#ident: burn::module::ModuleDisplayDefault
}
);
module_autodiff.add_predicate(
parse_quote! {