Proper differentiation of lambdas (no captures) & code cleanup

gojakuch · Nov 9, 2024 · 111dff3 · 111dff3
1 parent fa89545
commit 111dff3
Show file tree

Hide file tree

Showing 4 changed files with 305 additions and 15 deletions.
diff --git a/include/clad/Differentiator/Compatibility.h b/include/clad/Differentiator/Compatibility.h
@@ -403,6 +403,16 @@ getConstantArrayType(const ASTContext& Ctx, QualType EltTy,
    #define CLAD_COMPAT_CLANG15_Declarator_DeclarationAttrs_ExtraParam clang::ParsedAttributesView::none(),
 #endif
 
+#if CLANG_VERSION_MAJOR > 12
+#define CLAD_COMPAT_CXXRecordDecl_CreateLambda_DependencyKind(                 \
+    LAMBDACXXRECORDDECL)                                                       \
+  LAMBDACXXRECORDDECL->getLambdaDependencyKind()
+#else
+#define CLAD_COMPAT_CXXRecordDecl_CreateLambda_DependencyKind(                 \
+    LAMBDACXXRECORDDECL)                                                       \
+  LAMBDACXXRECORDDECL->isDependentLambda()
+#endif
+
 // Clang 12 add one extra param (FPO) that we get from Node in Create method of:
 // ImplicitCastExpr, CStyleCastExpr, CXXStaticCastExpr and CXXFunctionalCastExpr
 

diff --git a/include/clad/Differentiator/ReverseModeVisitor.h b/include/clad/Differentiator/ReverseModeVisitor.h
@@ -357,6 +357,11 @@ namespace clad {
         llvm::SmallVectorImpl<clang::Expr*>& outputArgs,
         clang::Expr* CUDAExecConfig = nullptr);
 
+    clang::CXXRecordDecl*
+    diffLambdaCXXRecordDecl(const clang::CXXRecordDecl* Original);
+    clang::CXXMethodDecl*
+    DifferentiateCallOperatorIfLambda(const clang::CXXRecordDecl* RD);
+
   public:
     ReverseModeVisitor(DerivativeBuilder& builder, const DiffRequest& request);
     virtual ~ReverseModeVisitor();
@@ -383,6 +388,7 @@ namespace clad {
     DerivativeAndOverload DerivePullback();
     StmtDiff VisitArraySubscriptExpr(const clang::ArraySubscriptExpr* ASE);
     StmtDiff VisitBinaryOperator(const clang::BinaryOperator* BinOp);
+    StmtDiff VisitLambdaExpr(const clang::LambdaExpr* LE);
     StmtDiff VisitCallExpr(const clang::CallExpr* CE);
     virtual StmtDiff VisitCompoundStmt(const clang::CompoundStmt* CS);
     StmtDiff VisitConditionalOperator(const clang::ConditionalOperator* CO);