Skip to content

Commit ec3a788

Browse files
add EnzymeRegisterCallHandler and EnzymeGradientUtilsNewFromOriginal to C API header (#2224)
1 parent 5651636 commit ec3a788

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

enzyme/Enzyme/CApi.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ void EnzymeRegisterAllocationHandler(char *Name, CustomShadowAlloc AHandle,
340340
};
341341
}
342342

343-
void EnzymeRegisterCallHandler(char *Name,
343+
void EnzymeRegisterCallHandler(const char *Name,
344344
CustomAugmentedFunctionForward FwdHandle,
345345
CustomFunctionReverse RevHandle) {
346346
auto &pair = customCallHandlers[Name];

enzyme/Enzyme/CApi.h

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,13 @@ LLVMValueRef EnzymeCreatePrimalAndGradient(
222222
uint8_t *_overwritten_args, size_t overwritten_args_size,
223223
EnzymeAugmentedReturnPtr augmented, uint8_t AtomicAdd);
224224

225+
void EnzymeRegisterCallHandler(const char *Name,
226+
CustomAugmentedFunctionForward FwdHandle,
227+
CustomFunctionReverse RevHandle);
228+
229+
LLVMValueRef EnzymeGradientUtilsNewFromOriginal(GradientUtils *gutils,
230+
LLVMValueRef val);
231+
225232
#ifdef __cplusplus
226233
}
227234
#endif

0 commit comments

Comments
 (0)