Skip to content

Fix restart.redistribute(): better default handling for mxg, myg in get_processor_layout() #65

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Mar 14, 2022
Merged

Fix restart.redistribute(): better default handling for mxg, myg in get_processor_layout() #65

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a648684
Better default handling for mxg, myg in get_processor_layout()
johnomotani Mar 3, 2022
08ef2c5
Tidy up default-handling logic in get_processor_layout()
johnomotani Mar 14, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 12 additions & 1 deletion boutdata/processor_rearrange.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ class processor_layout(processor_layout_):
pass


def get_processor_layout(boutfile, has_t_dimension=True, mxg=2, myg=2):
def get_processor_layout(boutfile, has_t_dimension=True, mxg=None, myg=None):
"""Given a BOUT.restart.* or BOUT.dmp.* file (as a DataFile object),
return the processor layout for its data

Expand All @@ -56,6 +56,17 @@ def get_processor_layout(boutfile, has_t_dimension=True, mxg=2, myg=2):

"""

if mxg is None:
try:
mxg = boutfile["MXG"]
except KeyError:
mxg = 2
if myg is None:
try:
myg = boutfile["MYG"]
except KeyError:
myg = 2

nxpe = boutfile.read("NXPE")
nype = boutfile.read("NYPE")
npes = nxpe * nype
Expand Down