Skip to content

Commit 9ebb61d

Browse files
add EnzymeCreatePrimalAndGradient to CApi.h (#2215)
* add EnzymeCreatePrimalAndGradient to CApi.h * update EnzymeCreatePrimalAndGradient signature
1 parent 7bc73fa commit 9ebb61d

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

enzyme/Enzyme/CApi.h

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,16 @@ LLVMValueRef EnzymeCreateForwardDiff(
212212
uint8_t *_overwritten_args, size_t overwritten_args_size,
213213
EnzymeAugmentedReturnPtr augmented);
214214

215+
LLVMValueRef EnzymeCreatePrimalAndGradient(
216+
EnzymeLogicRef Logic, LLVMValueRef request_req, LLVMBuilderRef request_ip,
217+
LLVMValueRef todiff, CDIFFE_TYPE retType, CDIFFE_TYPE *constant_args,
218+
size_t constant_args_size, EnzymeTypeAnalysisRef TA, uint8_t returnValue,
219+
uint8_t dretUsed, CDerivativeMode mode, uint8_t runtimeActivity,
220+
unsigned width, uint8_t freeMemory, LLVMTypeRef additionalArg,
221+
uint8_t forceAnonymousTape, CFnTypeInfo typeInfo,
222+
uint8_t *_overwritten_args, size_t overwritten_args_size,
223+
EnzymeAugmentedReturnPtr augmented, uint8_t AtomicAdd);
224+
215225
#ifdef __cplusplus
216226
}
217227
#endif

0 commit comments

Comments
 (0)