Reikriam

ddot-test

Mar 15th, 2021 (edited)
120
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C 0.42 KB | None | 0 0
  1. #include <stdio.h>
  2. #include <cblas.h>
  3.  
  4. extern double __enzyme_autodiff(void*, double*, double*);
  5.  
  6. double ddot(double* m, double* n) {
  7.     return cblas_ddot(10, m, 1, n, 1);
  8. }
  9.  
  10. double manual_daxpy(double* m, double* n) {
  11.     return __enzyme_autodiff(ddot, m, n);
  12. }
  13.  
  14.  
  15. int main() {
  16.     double m[3] = {1, 2, 3};
  17.     double n[3] = {4, 5, 6};
  18.     printf("ddot = %f, derivative = %f", ddot(m, n), manual_daxpy(m, n));
  19. }
Add Comment
Please, Sign In to add comment