`-convert-gradient-to-llvm`¶

Perform a dialect conversion from Gradient to LLVM

Options¶

-use-generic-functions : Use generic allocation and deallocation functions instead of the classic 'malloc', 'aligned_alloc' and 'free' functions

Make the signatures of ForwardOp and ReverseOp compatibleto Enzyme.

Insert Func.CallOp for ForwardOp and ReverseOp.

Lower operations in the gradient dialect to core MLIR dialects.

code/dialects/Gradient/GradientPasses

Download Python script

Download Notebook

View on GitHub