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

simplify tests use compile now

Browse Source
This commit is contained in:
Awni Hannun
2024-01-16 22:06:19 -08:00
parent 1c3f82ca17
commit ed4d867092
5 changed files with 75 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/src/transforms.cpp
View File

@@ -836,8 +836,5 @@ void init_transforms(py::module_& m) {
// Register static Python object cleanup before the interpreter exits
auto atexit = py::module_::import("atexit");
atexit.attr("register")(py::cpp_function([]() {
detail::compile_clear();
tree_cache().clear();
}));
atexit.attr("register")(py::cpp_function([]() { tree_cache().clear(); }));
}