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

fix py gc edge case (#2079)

Browse Source
This commit is contained in:
Awni Hannun
2025-04-18 12:46:53 -07:00
committed by GitHub
parent b529515eb1
commit 55935ccae7
3 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/src/mlx_func.cpp
View File

@@ -16,12 +16,12 @@ struct gc_func {
};
int gc_func_tp_traverse(PyObject* self, visitproc visit, void* arg) {
Py_VISIT(Py_TYPE(self));
gc_func* w = (gc_func*)self;
Py_VISIT(w->func);
for (auto d : w->deps) {
Py_VISIT(d);
}
Py_VISIT(Py_TYPE(self));
return 0;
};