zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 17:31:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixes #918 bug in compile_tests (#919)

Browse Source
This commit is contained in:
Angelos Katharopoulos 2024-03-27 22:37:37 -07:00 committed by GitHub
parent bab5386306
commit c4fd0e5ede
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/compile_tests.cpp
View File

@ -214,9 +214,10 @@ TEST_CASE("test simplify multi output") {
// is respected in simplification // is respected in simplification
{ {
auto e = compile(multi_three)({})[0]; auto e = compile(multi_three)({})[0];
CHECK_EQ(e.inputs().size(), 4);
CHECK_EQ(e.inputs().at(0).id(), e.inputs().at(2).id());
CHECK_EQ(e.inputs().at(1).id(), e.inputs().at(3).id());
CHECK(array_equal(e, array({0.0f, 1.0f, 0.0f, 1.0f})).item<bool>()); CHECK(array_equal(e, array({0.0f, 1.0f, 0.0f, 1.0f})).item<bool>());
CHECK_EQ(e.inputs()[0].id(), e.inputs()[2].id());
CHECK_EQ(e.inputs()[1].id(), e.inputs()[3].id());
} }
set_compile_mode(CompileMode::enabled); set_compile_mode(CompileMode::enabled);
} }