Skip to content

Commit

Permalink
linting
Browse files Browse the repository at this point in the history
  • Loading branch information
shingjan committed Aug 11, 2021
1 parent 193709b commit 00d9476
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions python/tvm/auto_scheduler/measure.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,6 @@
import shutil
import tempfile
import multiprocessing
from multiprocessing.pool import ThreadPool
import logging

import tvm._ffi
Expand Down Expand Up @@ -668,6 +667,7 @@ def local_build_worker(args):

try:
res = _local_build_worker(inp, build_func, verbose)
# pylint: disable=broad-except
except Exception:
if verbose >= 1:
print(".E", end="", flush=True) # Build error
Expand Down Expand Up @@ -700,7 +700,7 @@ def local_builder_build(inputs, timeout, n_parallel, build_func="default", verbo
res : List[BuildResult]
The build results of these MeasureInputs.
"""
executor = PopenPoolExecutor()
executor = PopenPoolExecutor(n_parallel, timeout)
tuple_res = executor.map_with_error_catching(
local_build_worker,
[
Expand Down Expand Up @@ -1181,6 +1181,7 @@ def _rpc_run_worker(args):

try:
res = _rpc_run(*args)
# pylint: disable=broad-except
except Exception:
if verbose >= 1:
print("*E", end="") # Run error
Expand Down

0 comments on commit 00d9476

Please sign in to comment.