Commit dc60c2f2 authored by Somnath, Suhas's avatar Somnath, Suhas
Browse files

Removed import of mpi4py. Handled by pyUSID.Process

parent 4918b4a9
...@@ -16,14 +16,6 @@ from pyUSID.io.hdf_utils import write_main_dataset, create_results_group, create ...@@ -16,14 +16,6 @@ from pyUSID.io.hdf_utils import write_main_dataset, create_results_group, create
from pyUSID.io.write_utils import Dimension from pyUSID.io.write_utils import Dimension
from pyUSID import USIDataset from pyUSID import USIDataset
from .utils.giv_utils import do_bayesian_inference, bayesian_inference_on_period from .utils.giv_utils import do_bayesian_inference, bayesian_inference_on_period
try:
from mpi4py import MPI
if MPI.COMM_WORLD.Get_size() == 1:
# mpi4py available but NOT called via mpirun or mpiexec => single node
MPI = None
except ImportError:
# mpi4py not even present! Single node by default:
MPI = None
cap_dtype = np.dtype({'names': ['Forward', 'Reverse'], cap_dtype = np.dtype({'names': ['Forward', 'Reverse'],
'formats': [np.float32, np.float32]}) 'formats': [np.float32, np.float32]})
...@@ -302,7 +294,7 @@ class GIVBayesian(Process): ...@@ -302,7 +294,7 @@ class GIVBayesian(Process):
verbose=self.verbose) verbose=self.verbose)
if self.verbose: if self.verbose:
print('Rank {} finished processing forward sections. Now working on reverse sections....'.format(self.mpi_rank)) print('Rank {} finished processing forward sections. Now working on reverse sections'.format(self.mpi_rank))
self.forward_results = parallel_compute(rolled_raw_data[:, half_v_steps:], do_bayesian_inference, self.forward_results = parallel_compute(rolled_raw_data[:, half_v_steps:], do_bayesian_inference,
cores=self._cores, cores=self._cores,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment