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

Fix rank>0 thread exiting during SparseGP multiprocessing (#731)

Browse source 
If you run mpiexec -n 2 on any code to try to use MPI multiprocessing for SparseGPRegression, it crashes (see #618) because the rank>1 processes do not have ret defined.
This commit is contained in:
Yixin Lin 2022-04-17 12:25:17 -04:00 • committed by GitHub
parent bb1bc50886
commit 0a9893e839
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
GPy/core/sparse_gp_mpi.py
View file

@ -108,6 +108,7 @@ class SparseGP_MPI(SparseGP):
raise raise
self._fail_count += 1 self._fail_count += 1
elif flag==-1: elif flag==-1:
ret = None
break break
else: else:
self._IN_OPTIMIZATION_ = False self._IN_OPTIMIZATION_ = False