Skip to content
Snippets Groups Projects
Commit c286d744 authored by Theo Steininger's avatar Theo Steininger
Browse files

Removed explicit knowledge about MPI from power_indices.py

parent 3275f4f3
Branches
Tags
No related merge requests found
# -*- coding: utf-8 -*-
import sys
import numpy as np
from d2o import distributed_data_object,\
STRATEGIES as DISTRIBUTION_STRATEGIES
from nifty.config import dependency_injector as gdi
MPI = gdi['MPI']
class PowerIndices(object):
def __init__(self, domain, distribution_strategy,
......@@ -299,6 +297,9 @@ class PowerIndices(object):
global_pundex = np.empty_like(local_pundex)
# Store the individual pundices in the local_pundex array
local_pundex[temp_uniqued_pindex] = local_temp_pundex
# Extract the MPI module from the global_pindex d2o
MPI_name = global_pindex.comm.__class__.__module__
MPI = sys.modules(MPI_name)
# Use Allreduce to find the first occurences/smallest pundices
global_pindex.comm.Allreduce(local_pundex,
global_pundex,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment