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 Packages Projects Releases Wiki Activity

Cleanup using namespace alias

Browse Source
This commit is contained in:
Joona Havukainen
2025-07-07 18:25:57 -07:00
parent c7af3016eb
commit 3e885f583a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
benchmarks/cpp/single_ops.cpp
View File

@@ -193,8 +193,8 @@ void time_reductions() {
auto argmin_along_1 = [&a]() { return mx::argmin(a, 1, false); }; auto argmin_along_1 = [&a]() { return mx::argmin(a, 1, false); };
TIME(argmin_along_1); TIME(argmin_along_1);
auto indices = mlx::core::array({1}); auto indices = mx::array({1});
auto updates = mlx::core::reshape(mlx::core::array({NAN}), {1, 1, 1}); auto updates = mx::reshape(mx::array({NAN}), {1, 1, 1});
std::vector<int> axes{0}; std::vector<int> axes{0};
auto b = scatter(a, {indices}, updates, axes); auto b = scatter(a, {indices}, updates, axes);
mx::eval(b); mx::eval(b);