mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-14 10:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added PyUnstable_Module_SetGIL to PyInit_cpu_feature_guard

Browse Source
This commit is contained in:
vfdev-5 2024-08-20 00:10:17 +02:00
parent b82bcdc572
commit 6546c4810b
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
jaxlib/cpu_feature_guard.c
View File

@ -172,5 +172,12 @@ static struct PyModuleDef cpu_feature_guard_module = {
#endif
EXPORT_SYMBOL PyMODINIT_FUNC PyInit_cpu_feature_guard(void) {
return PyModule_Create(&cpu_feature_guard_module);
PyObject *module = PyModule_Create(&cpu_feature_guard_module);
if (module == NULL) {
return NULL;
}
#ifdef Py_GIL_DISABLED
PyUnstable_Module_SetGIL(module, Py_MOD_GIL_NOT_USED);
#endif
return module;
}