From bd14886c2d9bddcb62833e2f2f7d59b8470259ae Mon Sep 17 00:00:00 2001 From: Martin Bubel Date: Wed, 8 Nov 2023 18:54:08 +0100 Subject: [PATCH] re-add mpi_test__.py --- mpi_test__.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 mpi_test__.py diff --git a/mpi_test__.py b/mpi_test__.py new file mode 100644 index 00000000..a05f5cb8 --- /dev/null +++ b/mpi_test__.py @@ -0,0 +1,21 @@ + +import numpy as np +import GPy +from mpi4py import MPI +np.random.seed(123456) +comm = MPI.COMM_WORLD +N = 100 +x = np.linspace(-6., 6., N) +y = np.sin(x) + np.random.randn(N) * 0.05 +comm.Bcast(y) +data = np.vstack([x,y]) +#infr = GPy.inference.latent_function_inference.VarDTC_minibatch(mpi_comm=comm) +m = GPy.models.SparseGPRegression(data[:1].T,data[1:2].T,mpi_comm=comm) +m.optimize(max_iters=10) +if comm.rank==0: + print float(m.objective_function()) + m.inference_method.mpi_comm=None + m.mpi_comm=None + m._trigger_params_changed() + print float(m.objective_function()) + \ No newline at end of file