You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I'd say reserve is a reasonable function to provide a custom-derivative for @wsmoses , what would we need to register?
Augmented primal is just a reserve for the same additional on the shadow, and do nothing for the reverse?
For context, that's the definition:
Gett all names for relevant vector reserve calls (here e.g. _ZN5alloc3vec16Vec$LT$T$C$A$GT$7reserve17hfbf4bb7fa5035d73E).
2A) Call EnzymeRegisterCallHandler (?), using the name from 1)
2B) Pass a nullptr as CustomFunctionReverse since it shouldn't need anything in the reverse pass
2C) Pass a Builder and gutils as first and third argument to the CustomAugmentedFunctionForward
2D) Pass callinst to the reserve() call from 1) as the second arg to CustomAugmentedFunctionForward
2E) do something (??) with the last three arguments of CustomAugmentedFunctionForward
Activity
ZuseZ4 commentedon Dec 6, 2024
A best guess on what to do here:
2A) Call
EnzymeRegisterCallHandler(?), using the name from 1)2B) Pass a nullptr as
CustomFunctionReversesince it shouldn't need anything in the reverse pass2C) Pass a Builder and gutils as first and third argument to the
CustomAugmentedFunctionForward2D) Pass callinst to the reserve() call from 1) as the second arg to
CustomAugmentedFunctionForward2E) do something (??) with the last three arguments of
CustomAugmentedFunctionForward