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

Chore: Add default language in pre-commit and bump hooks (#1652)

Browse Source
This commit is contained in:
Nripesh Niketan
2024-12-06 15:54:29 +00:00
committed by GitHub
parent fc88fd9097
commit 3bb5b4a302
3 changed files with 45 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/ops_tests.cpp
View File

@@ -3045,13 +3045,17 @@ TEST_CASE("test divmod") {
// Check that we can still eval when one output goes out of scope
std::vector<array> out_holder;
{ out_holder.push_back(divmod(x, y)[0]); }
{
out_holder.push_back(divmod(x, y)[0]);
}
eval(out_holder);
CHECK_EQ(out_holder[0].item<float>(), 0.0);
// Check that we can still eval when the other output goes out of scope
out_holder.clear();
{ out_holder.push_back(divmod(x, y)[1]); }
{
out_holder.push_back(divmod(x, y)[1]);
}
eval(out_holder);
CHECK_EQ(out_holder[0].item<float>(), 1.0);
}