Module autodiff 

Re-exports

use std::ptr; 🔒

use rustc_ast::expand::autodiff_attrs::DiffActivity; 🔒

use rustc_ast::expand::autodiff_attrs::DiffMode; 🔒

use rustc_ast::expand::typetree::FncTree; 🔒

use rustc_attr_ir::RustcAutodiff; 🔒

use rustc_codegen_ssa::mir::IntrinsicResult; 🔒

use rustc_codegen_ssa::mir::operand::OperandRef; 🔒

use rustc_codegen_ssa::mir::operand::OperandValue; 🔒

use rustc_codegen_ssa::mir::place::PlaceValue; 🔒

use rustc_codegen_ssa::traits::BaseTypeCodegenMethods; 🔒

use rustc_codegen_ssa::traits::BuilderMethods; 🔒

use rustc_codegen_ssa::traits::ReturnSlot; 🔒

use rustc_data_structures::thin_vec::ThinVec; 🔒

use rustc_middle::ty; 🔒

use rustc_middle::ty::Ty; 🔒

use rustc_middle::ty::TyCtxt; 🔒

use rustc_middle::ty::TypingEnv; 🔒

use rustc_span::bug; 🔒

use rustc_target::callconv::PassMode; 🔒

use tracing::debug; 🔒

use crate::builder::Builder; 🔒

use crate::builder::UNNAMED; 🔒

use crate::context::SimpleCx; 🔒

use crate::declare::declare_simple_fn; 🔒

use crate::llvm; 🔒

use crate::llvm::TRUE; 🔒

use crate::llvm::Type; 🔒

use crate::llvm::TypeKind; 🔒

use crate::llvm::Value; 🔒

Functions

adjust_activity_to_abi 🔒

generate_enzyme_call 🔒

When differentiating fn_to_diff, take a outer_fn and generate another function with expected naming and calling conventions which will be discovered by the enzyme LLVM pass and its body populated with the differentiated fn_to_diff. outer_fn is then modified to have a call to the generated function and handle the differences between the Rust calling convention and Enzyme.

match_args_from_caller_to_enzyme 🔒