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

WIP (tests)

Browse Source
This commit is contained in:
Ronan Collobert
2025-10-31 09:39:38 -07:00
parent 1bac0db7e3
commit 5baa361779
4 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/autograd_tests.cpp
View File

@@ -145,7 +145,7 @@ TEST_CASE("test jvp") {
// No dependence between input and output
{
auto fun = [](array in) { return array({1.0, 1.0}); };
auto fun = [](array /* in */) { return array({1.0, 1.0}); };
auto out = jvp(fun, array(1.0f), array(1.0f)).second;
CHECK(array_equal(out, zeros({2})).item<bool>());
}
@@ -195,7 +195,7 @@ TEST_CASE("test vjp") {
// No dependence between input and output
{
auto fun = [](array in) { return array(1.); };
auto fun = [](array /* in */) { return array(1.); };
auto out = vjp(fun, zeros({2}), array(1.)).second;
CHECK(array_equal(out, zeros({2})).item<bool>());
}