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

fix warnings showing up with -Wall (#2692)

Browse Source
This commit is contained in:
Ronan Collobert
2025-10-24 11:43:35 -07:00
committed by GitHub
parent 233384161e
commit 8f8af61a37
30 changed files with 35 additions and 222 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/transforms.cpp
View File

@@ -916,7 +916,7 @@ std::function<array(const array&, const array&)> vmap(
int in_axis_b /* = 0 */,
int out_axis /* = 0 */) {
auto vfun = vmap(
[in_axis_a, in_axis_b, out_axis, fun](const std::vector<array>& inputs) {
[fun](const std::vector<array>& inputs) {
return std::vector<array>{fun(inputs[0], inputs[1])};
},
{in_axis_a, in_axis_b},
@@ -929,7 +929,7 @@ std::function<array(const array&)> vmap(
int in_axis /* = 0 */,
int out_axis /* = 0 */) {
auto vfun = vmap(
[in_axis, out_axis, fun](const std::vector<array>& inputs) {
[fun](const std::vector<array>& inputs) {
return std::vector<array>{fun(inputs[0])};
},
{in_axis},