Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
ift
NIFTy
Commits
c286d744
Commit
c286d744
authored
Jan 11, 2017
by
Theo Steininger
Browse files
Removed explicit knowledge about MPI from power_indices.py
parent
3275f4f3
Changes
1
Hide whitespace changes
Inline
Side-by-side
nifty/spaces/power_space/power_indices.py
View file @
c286d744
# -*- 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
,
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment