Skip to content

Commit 5989d49

Browse files
authored
Add vector width getter capi (rust-lang#615)
1 parent aad739c commit 5989d49

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

enzyme/Enzyme/CApi.cpp

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -293,6 +293,19 @@ void EnzymeRegisterFwdCallHandler(char *Name, CustomFunctionForward FwdHandle) {
293293
};
294294
}
295295

296+
uint64_t EnzymeGradientUtilsGetWidth(GradientUtils *gutils) {
297+
return gutils->getWidth();
298+
}
299+
300+
LLVMTypeRef EnzymeGradientUtilsGetShadowType(GradientUtils *gutils,
301+
LLVMTypeRef T) {
302+
return wrap(gutils->getShadowType(unwrap(T)));
303+
}
304+
305+
LLVMTypeRef EnzymeGetShadowType(uint64_t width, LLVMTypeRef T) {
306+
return wrap(GradientUtils::getShadowType(unwrap(T), width));
307+
}
308+
296309
LLVMValueRef EnzymeGradientUtilsNewFromOriginal(GradientUtils *gutils,
297310
LLVMValueRef val) {
298311
return wrap(gutils->getNewFromOriginal(unwrap(val)));

0 commit comments

Comments
 (0)