From 9750f7924514c4eded3278bd6f9ee68830f567f7 Mon Sep 17 00:00:00 2001
From: James Hensman <james.hensman@gmail.com>
Date: Thu, 14 May 2015 17:40:10 +0100
Subject: [PATCH] omp for gradients_X

---
 GPy/kern/_src/stationary_utils.c | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/GPy/kern/_src/stationary_utils.c b/GPy/kern/_src/stationary_utils.c
index abc20820..2ebeae3c 100644
--- a/GPy/kern/_src/stationary_utils.c
+++ b/GPy/kern/_src/stationary_utils.c
@@ -1,19 +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;
+int n,d,nd,m;
+#pragma omp parallel for private(nd,n,d, retnd, m)
+for(nd=0;nd<(D*N);nd++){
+  n = nd/D;
+  d = nd%D;
+  retnd = 0.0;
+  for(m=0;m<M;m++){
+    retnd += tmp[n*M+m]*(X[nd]-X2[m*D+d]);
   }
+  grad[nd] = retnd;
 }
 } //grad_X
 
 
+
 void _lengthscale_grads(int N, int M, int Q, double* tmp, double* X, double* X2, double* grad){
 int n,m,q;
 double gradq, dist;