wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-01 03:09:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Store more useful information in jobs.

Browse Source
This commit is contained in:
Brendan Smithyman
2015-07-02 14:25:45 -04:00
parent 12e64f418c
commit 3f02ce9073
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Parallel.py
+6 -8
View File
@@ -200,7 +200,7 @@ class SystemSolver(object):
rank = ids[i]
if tag in systems:
relIDs.append(i)
relIDs.append(rank)
systemJobs = []
endNodes[tag] = []
@@ -214,7 +214,7 @@ class SystemSolver(object):
systemJobs.append(job)
label = 'Compute: %d, %d, %d'%(tag[0], tag[1], iworks)
systemNodes.append(label)
G.add_node(label, jobs=[job])
G.add_node(label, jobs=[job], subslice=work, tag=tag)
G.add_edge(tagNode, label)
iworks += 1
@@ -223,9 +223,7 @@ class SystemSolver(object):
for label in systemNodes:
G.add_edge(label, tagNode)
for i in relIDs:
rank = ids[i]
for rank in relIDs:
with lview.temp_flags(block=False, after=systemJobs):
# TODO: Remove dependency on self._hasSystemRank, once the SuperReferences
@@ -233,14 +231,14 @@ class SystemSolver(object):
# correct (allowed) systems.
job = lview.apply(clearRef, Reference(self.remote.endpointName), tag, rank)
clearJobs.append(job)
label = 'Wrap: %d, %d, %d'%(tag[0],tag[1], i)
G.add_node(label, jobs=[job])
label = 'Wrap: %d, %d, %d'%(tag[0],tag[1], rank)
G.add_node(label, jobs=[job], tag=tag, rank=rank)
endNodes[tag].append(label)
G.add_edge(tagNode, label)
else:
for i, sjob in enumerate(systemJobs):
with lview.temp_flags(block=False, follow=sjob):
with lview.temp_flags(block=False, follow=sjob, after=sjob):
job = lview.apply(clearRef, Reference(self.remote.endpointName), tag)
clearJobs.append(job)
label = 'Wrap: %d, %d, %d'%(tag[0],tag[1],i)