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

Update pre-commit hooks (#984)

Browse Source
This commit is contained in:
Nripesh Niketan
2024-04-11 18:27:53 +04:00
committed by GitHub
parent 12d4507ee3
commit ffff671273
10 changed files with 37 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/custom_vjp_tests.cpp
View File

@@ -18,9 +18,7 @@ TEST_CASE("test simple custom vjp") {
fn,
[&](const std::vector<array>&,
const std::vector<array>&,
const std::vector<array>&) {
return std::vector<array>{one, one};
});
const std::vector<array>&) { return std::vector<array>{one, one}; });
auto [z, g] = vjp(fn, {x, y}, {one, one});
CHECK_EQ(z[0].item<float>(), 6.0f);