Try using weakref?

Author: savannahostrowski

Tools/jit/_llvm.py

@@ -7,6 +7,7 @@
 import shlex
 import subprocess
 import typing
+import weakref
 
 _LLVM_VERSION = 19
 _LLVM_VERSION_PATTERN = re.compile(rf"version\s+{_LLVM_VERSION}\.\d+\.\d+\S*\s+")
@@ -32,14 +33,16 @@ async def wrapper(
     return wrapper
 
 
-_CORES: asyncio.BoundedSemaphore | None = None
+_CORES_BY_LOOP: weakref.WeakKeyDictionary[
+    asyncio.AbstractEventLoop, asyncio.BoundedSemaphore
+] = weakref.WeakKeyDictionary()
 
 
 def _get_cores() -> asyncio.BoundedSemaphore:
-    global _CORES
-    if _CORES is None:
-        _CORES = asyncio.BoundedSemaphore(os.cpu_count() or 1)
-    return _CORES
+    loop = asyncio.get_running_loop()
+    if loop not in _CORES_BY_LOOP:
+        _CORES_BY_LOOP[loop] = asyncio.BoundedSemaphore(os.cpu_count())
+    return _CORES_BY_LOOP[loop]
 
 
 async def _run(tool: str, args: typing.Iterable[str], echo: bool = False) -> str | None:

Tools/jit/build.py

@@ -28,9 +28,10 @@
     args = parser.parse_args()
 
     if len(args.target) == 1:
-        args.target.debug = args.debug
-        args.target.verbose = args.verbose
-        args.target.build(pathlib.Path.cwd(), comment=comment, force=args.force)
+        target = args.target[0]
+        target.debug = args.debug
+        target.verbose = args.verbose
+        target.build(pathlib.Path.cwd(), comment=comment, force=args.force)
 
     else:
         # Build for multiple targets (e.g. universal2)

configure

@@ -2617,50 +2617,50 @@ AS_VAR_IF([ac_cv_gcc_compat], [yes], [
                UNIVERSAL_ARCH_FLAGS="-arch ppc -arch i386"
                LIPO_32BIT_FLAGS=""
                ARCH_RUN_32BIT=""
-	             ARCH_TRIPLES=`echo {ppc,i386}-apple-darwin`
+               ARCH_TRIPLES=`echo {ppc,i386}-apple-darwin`
                ;;
             64-bit)
                UNIVERSAL_ARCH_FLAGS="-arch ppc64 -arch x86_64"
                LIPO_32BIT_FLAGS=""
                ARCH_RUN_32BIT="true"
-	             ARCH_TRIPLES=`echo {ppc64,x86_64}-apple-darwin`
+               ARCH_TRIPLES=`echo {ppc64,x86_64}-apple-darwin`
                ;;
             all)
                UNIVERSAL_ARCH_FLAGS="-arch i386 -arch ppc -arch ppc64 -arch x86_64"
                LIPO_32BIT_FLAGS="-extract ppc7400 -extract i386"
                ARCH_RUN_32BIT="/usr/bin/arch -i386 -ppc"
-	             ARCH_TRIPLES=`echo {i386,ppc,ppc64,x86_64}-apple-darwin`
+               ARCH_TRIPLES=`echo {i386,ppc,ppc64,x86_64}-apple-darwin`
                ;;
             universal2)
                UNIVERSAL_ARCH_FLAGS="-arch arm64 -arch x86_64"
                LIPO_32BIT_FLAGS=""
                LIPO_INTEL64_FLAGS="-extract x86_64"
                ARCH_RUN_32BIT="true"
-	             ARCH_TRIPLES=`echo {aarch64,x86_64}-apple-darwin`
+               ARCH_TRIPLES=`echo {aarch64,x86_64}-apple-darwin`
                ;;
             intel)
                UNIVERSAL_ARCH_FLAGS="-arch i386 -arch x86_64"
                LIPO_32BIT_FLAGS="-extract i386"
                ARCH_RUN_32BIT="/usr/bin/arch -i386"
-	             ARCH_TRIPLES=`echo {i386,x86_64}-apple-darwin`
+               ARCH_TRIPLES=`echo {i386,x86_64}-apple-darwin`
                ;;
             intel-32)
                UNIVERSAL_ARCH_FLAGS="-arch i386"
                LIPO_32BIT_FLAGS=""
                ARCH_RUN_32BIT=""
-	             ARCH_TRIPLES=i386-apple-darwin
+               ARCH_TRIPLES=i386-apple-darwin
                ;;
             intel-64)
                UNIVERSAL_ARCH_FLAGS="-arch x86_64"
                LIPO_32BIT_FLAGS=""
                ARCH_RUN_32BIT="true"
-	             ARCH_TRIPLES=x86_64-apple-darwin
+               ARCH_TRIPLES=x86_64-apple-darwin
                ;;
             3-way)
                UNIVERSAL_ARCH_FLAGS="-arch i386 -arch ppc -arch x86_64"
                LIPO_32BIT_FLAGS="-extract ppc7400 -extract i386"
                ARCH_RUN_32BIT="/usr/bin/arch -i386 -ppc"
-	             ARCH_TRIPLES=`echo {i386,ppc,x86_64}-apple-darwin`
+               ARCH_TRIPLES=`echo {i386,ppc,x86_64}-apple-darwin`
                ;;
             *)
                AC_MSG_ERROR([proper usage is --with-universal-arch=universal2|32-bit|64-bit|all|intel|3-way])
@@ -2776,7 +2776,7 @@ AS_VAR_IF([jit_flags],
           [],
           [AS_VAR_APPEND([CFLAGS_NODIST], [" $jit_flags"])
            AS_VAR_SET([REGEN_JIT_COMMAND],
-                      ["\$(PYTHON_FOR_REGEN) \$(srcdir)/Tools/jit/build.py ${ARCH_TRIPLES:-host}"])
+                      ["\$(PYTHON_FOR_REGEN) \$(srcdir)/Tools/jit/build.py ${ARCH_TRIPLES:-$host}"])
            AS_VAR_SET([JIT_STENCILS_H], ["jit_stencils.h"])
            AS_VAR_IF([Py_DEBUG],
                      [true],