apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-11 21:12:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

get rid of mpi4py import

Browse source
This commit is contained in:
Zhenwen Dai 2016-02-09 16:48:23 +00:00
parent 0d3f5a3e74
commit 0dd52981d0
1 changed files with 5 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

13
GPy/util/gpu_init.py
View file

@ -10,15 +10,12 @@ gpu_device = None
gpu_context = None
MPI_enabled = False
try:
from mpi4py import MPI
MPI_enabled = True
except:
pass
def initGPU():
try:
from mpi4py import MPI
MPI_enabled = True
except:
pass
try:
if MPI_enabled and MPI.COMM_WORLD.size>1:
from .parallel import get_id_within_node