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

Cycle leak break (#1856)

Browse Source 
* detect and break leaks in custom function

* detect and break leaks in custom function
This commit is contained in:
Awni Hannun
2025-02-11 14:45:02 -08:00
committed by GitHub
parent 142b77751d
commit 2a45056ba8
10 changed files with 396 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/src/mlx.cpp
View File

@@ -7,6 +7,7 @@
namespace nb = nanobind;
void init_mlx_func(nb::module_&);
void init_array(nb::module_&);
void init_device(nb::module_&);
void init_stream(nb::module_&);
@@ -28,6 +29,7 @@ NB_MODULE(core, m) {
nb::module_::import_("mlx._os_warning");
nb::set_leak_warnings(false);
init_mlx_func(m);
init_device(m);
init_stream(m);
init_array(m);