diff --git a/GPy/kern/_src/stationary.py b/GPy/kern/_src/stationary.py index 865086ac..443871af 100644 --- a/GPy/kern/_src/stationary.py +++ b/GPy/kern/_src/stationary.py @@ -249,25 +249,34 @@ class Stationary(Kern): X2 = X code = """ - int n,q,d; + int n,m,d; double retnd; - for(n=0;n + #include + """ + weave_options = {'headers' : [''], + 'extra_compile_args': ['-fopenmp -O3'], # -march=native'], + 'extra_link_args' : ['-lgomp']} + weave.inline(code, ['ret', 'N', 'D', 'M', 'tmp', 'X', 'X2'], type_converters=weave.converters.blitz, support_code=support_code, **weave_options) return ret/self.lengthscale**2 def gradients_X_diag(self, dL_dKdiag, X):