def run(self, loop, mapPlugin):
import joblib
with joblib.Parallel(n_jobs=self.ctx.params.cpu_count) as parallel:
ctxList = parallel(joblib.delayed(LoopWrapper(loop, True), False)(ctx) for ctx in mapPlugin.getWorkload())
timingCollection = TimingCollection(str(loop))
for ctx in ctxList:
for timing in ctx.timings:
timingCollection.addTiming(timing)
self.ctx.timings.append(timingCollection)
return ctxList
def test_join_timings_one_element(self):
ctx = Struct()
timing = SimpleTiming("Test", 100)
timings = TimingCollection(timing.name)
timings.duration = timing.duration
timings.addTiming(timing)
plugin = show_summary_stats.Plugin(ctx)
str_timing = plugin._join_timings(timings)
assert str_timing == "Test : 100.000 s"
def test_join_timings_one_element(self):
ctx = Struct()
timing = SimpleTiming("Test", 100)
timings= TimingCollection(timing.name)
timings.duration = timing.duration
timings.addTiming(timing)
plugin = show_summary_stats.Plugin(ctx)
str_timing = plugin._join_timings(timings)
assert str_timing == "Test : 100.000 s"
def run(self, loop, mapPlugin):
pool = Pool(self.ctx.params.cpu_count)
try:
ctxList = pool.map(LoopWrapper(loop, True), mapPlugin.getWorkload())
timingCollection = TimingCollection(str(loop))
for ctx in ctxList:
for timing in ctx.timings:
timingCollection.addTiming(timing)
self.ctx.timings.append(timingCollection)
return ctxList
finally:
pool.close()
def run(self, loop, mapPlugin):
import joblib
with joblib.Parallel(n_jobs=self.ctx.params.cpu_count) as parallel:
ctxList = parallel(
joblib.delayed(LoopWrapper(loop, True), False)(ctx)
for ctx in mapPlugin.getWorkload())
timingCollection = TimingCollection(str(loop))
for ctx in ctxList:
for timing in ctx.timings:
timingCollection.addTiming(timing)
self.ctx.timings.append(timingCollection)
return ctxList
def run(self, loop, mapPlugin):
pool = Pool(self.ctx.params.cpu_count)
try:
ctxList = pool.map(LoopWrapper(loop, True),
mapPlugin.getWorkload())
timingCollection = TimingCollection(str(loop))
for ctx in ctxList:
for timing in ctx.timings:
timingCollection.addTiming(timing)
self.ctx.timings.append(timingCollection)
return ctxList
finally:
pool.close()
def _reorder_timings(self, timings):
timing_map = OrderedDict()
keyword = ParallelPluginCollection([], 0, None).__str__()
for timing in timings:
if timing.name is not keyword:
try:
timing_map[timing.name].addTiming(timing)
except KeyError:
timing_map[timing.name] = TimingCollection(timing.name)
timing_map[timing.name].duration = timing.duration
timing_map[timing.name].addTiming(timing)
return timing_map