zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix slice (#2758)

Browse Source
This commit is contained in:
Awni Hannun
2025-11-13 11:30:02 -08:00
committed by GitHub
parent 8973550ff3
commit 66519fb348
6 changed files with 39 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/ops_tests.cpp
View File

@@ -292,7 +292,7 @@ TEST_CASE("test slice") {
out = slice(x, {0}, {4}, {2});
eval(out);
CHECK_EQ(out.data_size(), 4);
CHECK_EQ(out.data_size(), 3);
x = ones({4, 4});
out = slice(x, {0, 0}, {2, 4});
@@ -325,6 +325,20 @@ TEST_CASE("test slice") {
out = slice(x, {2, 2, 2}, {3, 4, 3});
eval(out);
CHECK_EQ(out.data_size(), 5);
x = ones({8});
out = slice(x, {7}, {-9}, {-1});
eval(out);
CHECK_EQ(out.data_size(), 8);
out = slice(x, {7}, {-9}, {-1});
eval(out);
CHECK_EQ(out.data_size(), 8);
x = ones({4, 2});
out = slice(x, {3, 0}, {-5, 2}, {-1, 1});
eval(out);
CHECK_EQ(out.data_size(), 8);
}
TEST_CASE("test slice update") {