apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-06-05 14:55:15 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

simplify the interface of using mpi

Browse source
This commit is contained in:
Zhenwen Dai 2014-07-02 10:52:47 +01:00
parent 216de32c0c
commit 1c165db845
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
GPy/models/bayesian_gplvm.py
View file

@ -57,11 +57,13 @@ class BayesianGPLVM(SparseGP):
if np.any(np.isnan(Y)): if np.any(np.isnan(Y)):
from ..inference.latent_function_inference.var_dtc import VarDTCMissingData from ..inference.latent_function_inference.var_dtc import VarDTCMissingData
inference_method = VarDTCMissingData() inference_method = VarDTCMissingData()
elif mpi_comm != None: elif mpi_comm is not None:
inference_method = VarDTC_minibatch(mpi_comm=mpi_comm) inference_method = VarDTC_minibatch(mpi_comm=mpi_comm)
else: else:
from ..inference.latent_function_inference.var_dtc import VarDTC from ..inference.latent_function_inference.var_dtc import VarDTC
inference_method = VarDTC() inference_method = VarDTC()
if isinstance(inference_method,VarDTC_minibatch):
inference_method.mpi_comm = mpi_comm
if kernel.useGPU and isinstance(inference_method, VarDTC_GPU): if kernel.useGPU and isinstance(inference_method, VarDTC_GPU):
kernel.psicomp.GPU_direct = True kernel.psicomp.GPU_direct = True