Re-exports§
use std::str::FromStr;🔒use std::string::String;🔒use rustc_ast::expand::autodiff_attrs::DiffActivity;🔒use rustc_ast::expand::autodiff_attrs::DiffMode;🔒use rustc_ast::expand::autodiff_attrs::valid_input_activity;🔒use rustc_ast::expand::autodiff_attrs::valid_ret_activity;🔒use rustc_ast::expand::autodiff_attrs::valid_ty_for_activity;🔒use rustc_ast::token::IdentKind;🔒use rustc_ast::token::Lit;🔒use rustc_ast::token::LitKind;🔒use rustc_ast::token::Token;🔒use rustc_ast::token::TokenKind;🔒use rustc_ast as ast;🔒use rustc_ast::FnRetTy;🔒use rustc_ast::FnSig;🔒use rustc_ast::GenericArg;🔒use rustc_ast::GenericParamKind;🔒use rustc_ast::Generics;🔒use rustc_ast::ItemKind;🔒use rustc_ast::MetaItemInner;🔒use rustc_ast::PatKind;🔒use rustc_ast::TyKind;🔒use rustc_ast::Visibility;🔒use rustc_attr_ir::RustcAutodiff;🔒use rustc_expand::base::Annotatable;🔒use rustc_expand::base::ExtCtxt;🔒use rustc_span::DUMMY_SP;🔒use rustc_span::Ident;🔒use rustc_span::Span;🔒use rustc_span::Symbol;🔒use rustc_span::kw;🔒use rustc_span::sym;🔒use thin_vec::ThinVec;🔒use thin_vec::thin_vec;🔒use tracing::debug;🔒use tracing::trace;🔒use crate::diagnostics;🔒use rustc_ast::tokenstream::*;🔒use rustc_ast::visit::AssocCtxt::*;🔒
Functions§
- assure_
mut_ 🔒ref - call_
autodiff 🔒 - expand_
forward 🔒 - expand_
reverse 🔒 - expand_
with_ 🔒mode - We expand the autodiff macro to generate a new placeholder function which passes type-checking and can be called by users. The exact signature of the generated function depends on the configuration provided by the user, but here is an example:
- extract_
item_ 🔒info - first_
ident 🔒 - from_
ast 🔒 - gen_
enzyme_ 🔒decl - gen_
turbofish_ 🔒expr - has_ret 🔒
- meta_
item_ 🔒inner_ to_ ts - name 🔒
- outer_
normal_ 🔒attr - width 🔒