Lerking/OSACA
1
0
Fork 0
mirror of https://github.com/RRZE-HPC/OSACA.git synced 2026-01-05 18:50:08 +01:00
Code Issues Packages Projects Releases Wiki Activity

added no timeout option

Browse Source
This commit is contained in:
JanLJL
2021-04-19 10:57:51 +02:00
parent cfc061e5e3
commit 3f31235f8a
2 changed files with 22 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
osaca/semantics/kernel_dg.py
View File

@@ -2,12 +2,10 @@
import copy
import time
from collections import defaultdict
from itertools import accumulate, chain, product
from itertools import chain
from multiprocessing import Manager, Process, cpu_count
import networkx as nx
from osaca.parser import AttrDict
from osaca.semantics import INSTR_FLAGS, ArchSemantics, MachineModel
@@ -123,22 +121,27 @@ class KernelDG(nx.DiGraph):
]
for p in processes:
p.start()
start_time = time.time()
while time.time() - start_time <= timeout:
if any(p.is_alive() for p in processes):
time.sleep(0.2)
else:
# all procs done
for p in processes:
p.join()
break
else:
self.timed_out = True
# terminate running processes
if (timeout == -1):
# no timeout
for p in processes:
if p.is_alive():
p.kill()
p.join()
else:
start_time = time.time()
while time.time() - start_time <= timeout:
if any(p.is_alive() for p in processes):
time.sleep(0.2)
else:
# all procs done
for p in processes:
p.join()
break
else:
self.timed_out = True
# terminate running processes
for p in processes:
if p.is_alive():
p.kill()
p.join()
all_paths = list(all_paths)
else:
# sequential execution to avoid overhead when analyzing smaller kernels