zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-03 18:18:15 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

basic compile scaffold works

Browse Source
This commit is contained in:
Awni Hannun
2024-01-14 09:50:31 -08:00
parent 264e9ad57e
commit 21062680d5
4 changed files with 234 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/compile_tests.cpp
View File

@@ -14,4 +14,7 @@ TEST_CASE("test simple compile") {
auto compfn = compile(simple_fun);
auto out = compfn({array(1.0), array(2.0)})[0];
CHECK_EQ(out.item<float>(), 3.0f);
out = compfn({array(1.0), array(2.0)})[0];
CHECK_EQ(out.item<float>(), 3.0f);
}