rust-lang/rust
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

support both trait and non-trail associated declaration

Browse Source
This commit is contained in:
bit-aloo
2025-04-21 13:22:56 +05:30
parent b8005bff32
commit c0c6d4280c
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
compiler/rustc_builtin_macros/src/autodiff.rs
View File

@@ -217,14 +217,12 @@ mod llvm_enzyme {
ast::StmtKind::Item(iitem) => extract_item_info(iitem), ast::StmtKind::Item(iitem) => extract_item_info(iitem),
_ => None, _ => None,
}, },
Annotatable::AssocItem(assoc_item, Impl { of_trait: false }) => { Annotatable::AssocItem(assoc_item, Impl { .. }) => match &assoc_item.kind {
match &assoc_item.kind { ast::AssocItemKind::Fn(box ast::Fn { sig, ident, .. }) => {
ast::AssocItemKind::Fn(box ast::Fn { sig, ident, .. }) => { Some((assoc_item.vis.clone(), sig.clone(), ident.clone()))
Some((assoc_item.vis.clone(), sig.clone(), ident.clone()))
}
_ => None,
} }
} _ => None,
},
_ => None, _ => None,
}) else { }) else {
dcx.emit_err(errors::AutoDiffInvalidApplication { span: item.span() }); dcx.emit_err(errors::AutoDiffInvalidApplication { span: item.span() });
@@ -365,7 +363,7 @@ mod llvm_enzyme {
} }
Annotatable::Item(iitem.clone()) Annotatable::Item(iitem.clone())
} }
Annotatable::AssocItem(ref mut assoc_item, i @ Impl { of_trait: false }) => { Annotatable::AssocItem(ref mut assoc_item, i @ Impl { .. }) => {
if !assoc_item.attrs.iter().any(|a| same_attribute(&a.kind, &attr.kind)) { if !assoc_item.attrs.iter().any(|a| same_attribute(&a.kind, &attr.kind)) {
assoc_item.attrs.push(attr); assoc_item.attrs.push(attr);
} }