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

add mpi support for sparsegpregression

Browse source
This commit is contained in:
Zhenwen Dai 2014-11-05 17:50:23 +00:00
parent 1dbe3e34b0
commit f5377aa441
4 changed files with 46 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

3
GPy/core/sparse_gp_mpi.py
View file

@ -47,7 +47,7 @@ class SparseGP_MPI(SparseGP):
if variational_prior is not None:
self.link_parameter(variational_prior)
self.mpi_comm = mpi_comm
# Manage the data (Y) division
if mpi_comm != None:
@ -60,7 +60,6 @@ class SparseGP_MPI(SparseGP):
mpi_comm.Bcast(self.param_array, root=0)
self.update_model(True)
def __getstate__(self):
dc = super(SparseGP_MPI, self).__getstate__()
dc['mpi_comm'] = None