apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-12 05:22: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

5
GPy/util/gpu_init.py
View file

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