-
Notifications
You must be signed in to change notification settings - Fork 6
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.
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
_match_messages #131
base: develop
Are you sure you want to change the base?
_match_messages #131
Conversation
pipit/trace.py
Outdated
matching_events = list(self.events["_matching_event"]) | ||
matching_times = list(self.events["_matching_timestamp"]) | ||
|
||
mpi_events = self.events[ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should rename to comm_events
or some equivalent
pipit/trace.py
Outdated
matching_times = list(self.events["_matching_timestamp"]) | ||
|
||
mpi_events = self.events[ | ||
self.events["Name"].isin(["MpiSend", "MpiRecv", "MpiIsend", "MpiIrecv"]) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Update to check if the event name is in send_events_names + receive_events_names
pipit/trace.py
Outdated
self.events["Name"].isin(["MpiSend", "MpiRecv", "MpiIsend", "MpiIrecv"]) | ||
] | ||
|
||
queue = [[] for _ in range(len(self.events["Process"].unique()))] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Update queue to be a dict per process, and index the dict per receiver process with sender processes. Also comments should be added to explain this.
pipit/trace.py
Outdated
processes = list(mpi_events["Process"]) | ||
|
||
# Iterate through all events | ||
for i in range(len(mpi_events)): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Change to iterate over send events and receive events separately, so that if a process with a receive event is iterated over, we can confirm that the send event has already been seen and added to the queue.
pipit/trace.py
Outdated
if curr_name == "MpiSend" or curr_name == "MpiIsend": | ||
# Add current dataframe index, timestmap, and process to stack | ||
if "receiver" in curr_attrs: | ||
queue[curr_attrs["receiver"]].append( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Update for new dict queue implementation.
pipit/trace.py
Outdated
|
||
# we want to iterate through the queue in order | ||
# until we find the corresponding "send" event | ||
while send_process != curr_attrs["sender"] and i < len( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Update for new dict queue implementation.
eac3cae
to
2ee3f03
Compare
I'm getting a key error when running on the Kripke 32-process trace. |
No description provided.