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

Fix empty array construction in cpp (#684)

Browse Source
This commit is contained in:
Awni Hannun
2024-02-13 23:34:17 -08:00
committed by GitHub
parent 0c65517e91
commit 1eb04aa23f
4 changed files with 29 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/array_tests.cpp
View File

@@ -591,3 +591,21 @@ TEST_CASE("test array shared buffer") {
eval(a + b);
}
TEST_CASE("test make empty array") {
auto a = array({});
CHECK_EQ(a.size(), 0);
CHECK_EQ(a.dtype(), float32);
a = array({}, int32);
CHECK_EQ(a.size(), 0);
CHECK_EQ(a.dtype(), int32);
a = array({}, float32);
CHECK_EQ(a.size(), 0);
CHECK_EQ(a.dtype(), float32);
a = array({}, bool_);
CHECK_EQ(a.size(), 0);
CHECK_EQ(a.dtype(), bool_);
}