Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <stdio.h>
- #include <cblas.h>
- extern double __enzyme_autodiff(void*, double*, double*);
- double ddot(double* m, double* n) {
- return cblas_ddot(10, m, 1, n, 1);
- }
- double manual_daxpy(double* m, double* n) {
- return __enzyme_autodiff(ddot, m, n);
- }
- int main() {
- double m[3] = {1, 2, 3};
- double n[3] = {4, 5, 6};
- printf("ddot = %f, derivative = %f", ddot(m, n), manual_daxpy(m, n));
- }
Add Comment
Please, Sign In to add comment