Skip to content

Commit 13c223b

Browse files
committed
Default to using self.cancellation_manager
Part of google#96
1 parent 390faa2 commit 13c223b

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

compiler_opt/rl/inlining/inlining_runner.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,9 @@ def compile_fn(
7171
cancelled work.
7272
RuntimeError: if llvm-size produces unexpected output.
7373
"""
74+
if cancellation_manager is None:
75+
cancellation_manager = self._cancellation_manager
76+
7477
working_dir = tempfile.mkdtemp()
7578

7679
log_path = os.path.join(working_dir, 'log')

compiler_opt/rl/regalloc/regalloc_runner.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,9 @@ def compile_fn(
6969
cancelled work.
7070
RuntimeError: if llvm-size produces unexpected output.
7171
"""
72+
if cancellation_manager is None:
73+
cancellation_manager = self._cancellation_manager
74+
7275
working_dir = tempfile.mkdtemp()
7376

7477
log_path = os.path.join(working_dir, 'log')

0 commit comments

Comments
 (0)