diff --git a/GPy/kern/_src/psi_comp/rbf_psi_gpucomp.py b/GPy/kern/_src/psi_comp/rbf_psi_gpucomp.py index 270801ff..43fffcd1 100644 --- a/GPy/kern/_src/psi_comp/rbf_psi_gpucomp.py +++ b/GPy/kern/_src/psi_comp/rbf_psi_gpucomp.py @@ -58,7 +58,23 @@ gpu_code = """ } } - __global__ void psi1computations(double *psi1, double var, double *l, double *Z, double *mu, double *S, int N, int M, int Q) + __global__ void compDenom(double *log_denom1, double *log_denom2, double *l, double *S, int N, int Q) + { + int n_start, n_end; + divide_data(N, gridDim.x, blockIdx.x, &n_start, &n_end); + + for(int i=n_start*Q+threadIdx.x; i