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

shapeless compile in docs and partially shapeless reshape (#1742)

Browse Source
This commit is contained in:
Awni Hannun
2025-01-02 16:24:42 -08:00
committed by GitHub
parent a64a8dfe45
commit ae69cb15e9
6 changed files with 137 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/compile_tests.cpp
View File

@@ -685,7 +685,8 @@ auto compile_shapeless_ok(const std::vector<array>& inputs) {
TEST_CASE("test shapeless compile") {
{
auto cfun = compile(compile_shapeless_not_ok, /* shapeless */ true);
CHECK_THROWS(cfun({array({1, 2, 3, 4})}));
cfun({array({1, 2, 3, 4})});
CHECK_THROWS(cfun({array({1, 2, 3, 4, 5})}));
}
{