CI: pin pixi asan build to 3.15.0a6 and re-introduce SetImmortal bindings (#30901)

ngoldbaum · web-flow · commit e87a3c5fef53 · 2026-02-27T13:16:32.000-07:00
diff --git a/numpy/_core/src/common/pythoncapi-compat b/numpy/_core/src/common/pythoncapi-compat
@@ -1 +1 @@
-Subproject commit 90c06a4cae557bdbfa4f231a781d2b5c1a8f6d1c
+Subproject commit 8636bccf29adfa23463f810b3c2830f7cff1e933
diff --git a/numpy/_core/src/multiarray/multiarraymodule.c b/numpy/_core/src/multiarray/multiarraymodule.c
@@ -5170,7 +5170,13 @@ _multiarray_umath_exec(PyObject *m) {
     if (PyDataMem_DefaultHandler == NULL) {
         return -1;
     }
-
+#ifdef Py_GIL_DISABLED
+    if (PyUnstable_SetImmortal(PyDataMem_DefaultHandler) == 0) {
+        PyErr_SetString(PyExc_RuntimeError,
+                        "Could not mark memory handler capsule as immortal");
+        return -1;
+    }
+#endif
     /*
      * Initialize the context-local current handler
      * with the default PyDataMem_Handler capsule.
diff --git a/numpy/_core/src/umath/extobj.c b/numpy/_core/src/umath/extobj.c
@@ -15,6 +15,7 @@
 #include "numpy/ufuncobject.h"
 
 #include "common.h"
+#include "npy_pycompat.h"
 
 
 #define UFUNC_ERR_IGNORE 0
@@ -145,6 +146,13 @@ init_extobj(void)
     if (npy_static_pydata.default_extobj_capsule == NULL) {
         return -1;
     }
+#ifdef Py_GIL_DISABLED
+    if (PyUnstable_SetImmortal(npy_static_pydata.default_extobj_capsule) == 0) {
+        PyErr_SetString(PyExc_RuntimeError, "Could not mark extobj capsule as immortal");
+        Py_CLEAR(npy_static_pydata.default_extobj_capsule);
+        return -1;
+    }
+#endif
     npy_static_pydata.npy_extobj_contextvar = PyContextVar_New(
             "numpy.ufunc.extobj", npy_static_pydata.default_extobj_capsule);
     if (npy_static_pydata.npy_extobj_contextvar == NULL) {
diff --git a/pixi-packages/asan/pixi.toml b/pixi-packages/asan/pixi.toml
@@ -19,6 +19,8 @@ extra-args = ["-Csetup-args=-Db_sanitize=address", "-Csetup-args=-Dbuildtype=deb
 [package.host-dependencies]
 python.git = "https://github.com/python/cpython"
 python.subdirectory = "Tools/pixi-packages/asan"
+# v3.15.0a6
+python.rev = "15b216f30d0445469ec31bc7509fcc55a216ef7c"
 
 meson-python = "*"
 cython = "*"
