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

contiguous op / prim (#1612)

Browse Source
This commit is contained in:
Awni Hannun
2024-11-21 19:51:49 -08:00
committed by GitHub
parent 0d5e7716ad
commit dcca0d7477
11 changed files with 104 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/ops_tests.cpp
View File

@@ -3747,3 +3747,17 @@ TEST_CASE("test roll") {
CHECK(array_equal(y, array({8, 9, 5, 6, 7, 3, 4, 0, 1, 2}, {2, 5}))
.item<bool>());
}
TEST_CASE("test contiguous") {
auto x = array({1, 2, 3});
x = contiguous(broadcast_to(x, {2, 2, 3}));
eval(x);
CHECK(x.flags().row_contiguous);
CHECK_EQ(x.strides(), decltype(x.strides()){6, 3, 1});
x = array({1, 2, 1, 2}, {2, 2});
x = contiguous(transpose(x), true);
eval(x);
CHECK(x.flags().col_contiguous);
CHECK_EQ(x.strides(), decltype(x.strides()){1, 2});
}