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

try dynamic reshape

Browse Source
This commit is contained in:
Awni Hannun
2024-12-06 12:09:08 -08:00
parent 40c62c1321
commit ee59d50293
7 changed files with 164 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/ops_tests.cpp
View File

@@ -3769,3 +3769,23 @@ TEST_CASE("test contiguous") {
CHECK(x.flags().col_contiguous);
CHECK_EQ(x.strides(), decltype(x.strides()){1, 2});
}
TEST_CASE("test dynamic reshape") {
auto x = array({1}, {1, 1, 1});
CHECK_EQ(dynamic_reshape(x, {}).shape(), Shape{});
// Bad character
CHECK_THROWS(dynamic_reshape(x, {"&", 1, 1}));
// No dim in string
CHECK_THROWS(dynamic_reshape(x, {"1", 1, 1}));
// Too many dims
CHECK_THROWS(dynamic_reshape(x, {"abcd", 1, 1}));
// Too many dims
CHECK_THROWS(dynamic_reshape(x, {"a", "b", "c", "d"}));
// Too many inferred dims
CHECK_THROWS(dynamic_reshape(x, {"a", -1, -1}));
}