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

fix compile merging (#2150)

Browse Source
This commit is contained in:
Awni Hannun
2025-05-02 15:08:50 -07:00
committed by GitHub
parent 481349495b
commit 9c5e7da507
2 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/compile_tests.cpp
View File

@@ -795,3 +795,12 @@ TEST_CASE("test compile lambda") {
out = cfun2({array(0)});
CHECK_EQ(out[0].item<int>(), 3);
}
TEST_CASE("test compile with no-ops") {
auto fun = [](const std::vector<array>& inputs) {
return std::vector<array>{abs(stop_gradient(abs(inputs[0])))};
};
auto in = array(1.0);
auto out = compile(fun)({in})[0];
CHECK_EQ(out.inputs()[0].id(), in.id());
}