Skip to content

Commit

Permalink
passes: implement multiprocessing depending on number of cpus
Browse files Browse the repository at this point in the history
  • Loading branch information
PhilippvK committed Mar 4, 2024
1 parent 6d39cb7 commit 386a8ed
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions seal5/passes.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import time
import multiprocessing
from pathlib import Path
from enum import Enum, IntFlag, auto
from dataclasses import dataclass
Expand All @@ -11,6 +12,10 @@
logger = get_logger()


# NUM_THREADS = multiprocessing.cpu_count()
NUM_THREADS = 1


class PassFormat(IntFlag):
NONE = auto()
CDSL = auto()
Expand Down Expand Up @@ -148,13 +153,14 @@ def __init__(
pass_list: List[Seal5Pass],
skip: Optional[List[str]] = None,
only: Optional[List[str]] = None,
parallel: int = 2,
parent: Optional["PassManager"] = None,
parallel: Optional[int] = None,
):
self.name = name
self.pass_list = pass_list
self.skip = skip if skip is not None else (parent.skip if parent else [])
self.only = only if only is not None else (parent.only if parent else [])
self.parallel = parallel
self.parallel = parallel if parallel is not None else (parent.parallel if parent else NUM_THREADS)
self.metrics: dict = {}
self.open: bool = False

Expand Down

0 comments on commit 386a8ed

Please sign in to comment.