zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-15 01:19:21 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

No gil eval (#565)

Browse Source
This commit is contained in:
Angelos Katharopoulos
2024-01-26 22:03:52 -08:00
committed by GitHub
parent 8993382aaa
commit 37d98ba6ff
3 changed files with 40 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/src/transforms.cpp
View File

@@ -509,7 +509,10 @@ void init_transforms(py::module_& m) {
"eval",
[](const py::args& args) {
std::vector<array> arrays = tree_flatten(args);
eval(arrays);
{
py::gil_scoped_release nogil;
eval(arrays);
}
},
R"pbdoc(
eval(*args) -> None