add EnzymeCreatePrimalAndGradient to CApi.h (#2215)

* add EnzymeCreatePrimalAndGradient to CApi.h * update EnzymeCreatePrimalAndGradient signature
EnzymeAD · Jan 3, 2025 · 9ebb61d · 9ebb61d
1 parent 7bc73fa
commit 9ebb61d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/enzyme/Enzyme/CApi.h b/enzyme/Enzyme/CApi.h
@@ -212,6 +212,16 @@ LLVMValueRef EnzymeCreateForwardDiff(
     uint8_t *_overwritten_args, size_t overwritten_args_size,
     EnzymeAugmentedReturnPtr augmented);
 
+LLVMValueRef EnzymeCreatePrimalAndGradient(
+    EnzymeLogicRef Logic, LLVMValueRef request_req, LLVMBuilderRef request_ip,
+    LLVMValueRef todiff, CDIFFE_TYPE retType, CDIFFE_TYPE *constant_args,
+    size_t constant_args_size, EnzymeTypeAnalysisRef TA, uint8_t returnValue,
+    uint8_t dretUsed, CDerivativeMode mode, uint8_t runtimeActivity,
+    unsigned width, uint8_t freeMemory, LLVMTypeRef additionalArg,
+    uint8_t forceAnonymousTape, CFnTypeInfo typeInfo,
+    uint8_t *_overwritten_args, size_t overwritten_args_size,
+    EnzymeAugmentedReturnPtr augmented, uint8_t AtomicAdd);
+
 #ifdef __cplusplus
 }
 #endif