From 885614489087650edb0bb2e1a1c99c65c47105e3 Mon Sep 17 00:00:00 2001 From: Zhenwen Dai Date: Fri, 24 Oct 2014 02:42:40 +0100 Subject: [PATCH] enable gpu for inference X --- GPy/inference/latent_function_inference/inference_X.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/GPy/inference/latent_function_inference/inference_X.py b/GPy/inference/latent_function_inference/inference_X.py index c34896d7..e3a3d251 100644 --- a/GPy/inference/latent_function_inference/inference_X.py +++ b/GPy/inference/latent_function_inference/inference_X.py @@ -25,6 +25,8 @@ class Inference_X(Model): super(Inference_X, self).__init__(name) self.likelihood = model.likelihood.copy() self.kern = model.kern.copy() + if model.kern.useGPU: + self.kern.GPU(True) from copy import deepcopy self.posterior = deepcopy(model.posterior) self.variational_prior = model.variational_prior.copy()