use Runner

Author: savannahostrowski

Tools/jit/_llvm.py

@@ -32,19 +32,12 @@ async def wrapper(
     return wrapper
 
 
-_CORES: asyncio.BoundedSemaphore | None = None
-
-
-def _get_cores() -> asyncio.BoundedSemaphore:
-    global _CORES
-    if _CORES is None:
-        _CORES = asyncio.BoundedSemaphore(os.cpu_count() or 1)
-    return _CORES
+_CORES = asyncio.BoundedSemaphore(os.cpu_count() or 1)
 
 
 async def _run(tool: str, args: typing.Iterable[str], echo: bool = False) -> str | None:
     command = [tool, *args]
-    async with _get_cores():
+    async with _CORES:
         if echo:
             print(shlex.join(command))
         try:

Tools/jit/_targets.py

@@ -25,6 +25,7 @@
 CPYTHON = TOOLS.parent
 PYTHON_EXECUTOR_CASES_C_H = CPYTHON / "Python" / "executor_cases.c.h"
 TOOLS_JIT_TEMPLATE_C = TOOLS_JIT / "template.c"
+ASYNCIO_RUNNER = asyncio.Runner()
 
 _S = typing.TypeVar("_S", _schema.COFFSection, _schema.ELFSection, _schema.MachOSection)
 _R = typing.TypeVar(
@@ -199,7 +200,7 @@ def build(
             and jit_stencils.read_text().startswith(digest)
         ):
             return
-        stencil_groups = asyncio.run(self._build_stencils())
+        stencil_groups = ASYNCIO_RUNNER.run(self._build_stencils())
         jit_stencils_new = out / "jit_stencils.h.new"
         try:
             with jit_stencils_new.open("w") as file: