Merge pull request #212 from volpepe/patch-2

Kill process after timeout expired

Author: ChrisCummins

programl/create_ops.py

@@ -167,6 +167,7 @@ def _run_one(src: str):
         try:
             stdout, stderr = process.communicate(src.encode("utf-8"), timeout=timeout)
         except subprocess.TimeoutExpired as e:
+            process.kill()
             raise TimeoutError(str(e)) from e
 
         return _graph_from_subprocess(process, stdout, stderr)
@@ -268,6 +269,7 @@ def _run_one(one_args):
         try:
             stdout, stderr = process.communicate(timeout=timeout)
         except subprocess.TimeoutExpired as e:
+            process.kill()
             raise TimeoutError(str(e)) from e
 
         return _graph_from_subprocess(process, stdout, stderr)
@@ -355,6 +357,7 @@ def _run_one(ir: str):
         try:
             stdout, stderr = process.communicate(ir.encode("utf-8"), timeout=timeout)
         except subprocess.TimeoutExpired as e:
+            process.kill()
             raise TimeoutError(str(e)) from e
 
         return _graph_from_subprocess(process, stdout, stderr)
@@ -416,6 +419,7 @@ def _run_one(hlo: HloProto) -> ProgramGraph:
                 hlo.SerializeToString(), timeout=timeout
             )
         except subprocess.TimeoutExpired as e:
+            process.kill()
             raise TimeoutError(str(e)) from e
 
         return _graph_from_subprocess(process, stdout, stderr)