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

Xcode 160 (#1384)

Browse Source 
* xcode 16.0 with debug tests

* limit nproc for builds

* vmap bug

* assert bug

* run python tests in debug mode

* fix view, bool copies preserve bits'

* actual view fix
This commit is contained in:
Awni Hannun
2024-09-10 15:15:17 -07:00
committed by GitHub
parent e7e59c6f05
commit 02efb310ca
5 changed files with 45 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/src/transforms.cpp
View File

@@ -803,7 +803,7 @@ class PyCustomFunction {
"[custom vmap] Function should only accept positional arguments");
}
int arr_index;
int arr_index = 0;
auto new_axes =
nb::cast<nb::tuple>(tree_map(args, [&](nb::handle element) {
int axis = axes[arr_index++];