initial cython commit

This commit is contained in:
James Hensman 2015-04-27 14:21:46 +01:00
parent be40318e0b
commit b36a845821
6 changed files with 5787 additions and 1 deletions

View file

@ -0,0 +1,20 @@
void _grad_X(int N, int D, int M, double* X, double* X2, double* tmp, double* grad){
int n,m,d;
double retnd;
#pragma omp parallel for private(n,d, retnd, m)
for(d=0;d<D;d++){
for(n=0;n<N;n++){
retnd = 0.0;
for(m=0;m<M;m++){
retnd += tmp[n*M+m]*(X[n*D+d]-X2[m*D+d]);
}
grad[n*D+d] = retnd;
}
}
} //grad_X
//#weave_options = {'headers' : ['<omp.h>'],
//'extra_compile_args': ['-fopenmp -O3'], # -march=native'],
//'extra_link_args' : ['-lgomp']}