mirror of
https://github.com/ml-explore/mlx.git
synced 2025-10-20 09:33:16 +08:00
Use int64 stride everywhere (#1671)
* use int64 stride everywhere * fix ext * fix ext * more shape + cleanup * one more * few more
This commit is contained in:
@@ -16,7 +16,7 @@ TEST_CASE("test matmul") {
|
||||
a = array({1.0});
|
||||
b = array({1.0});
|
||||
auto out = matmul(a, b);
|
||||
CHECK_EQ(out.shape(), std::vector<int>{});
|
||||
CHECK_EQ(out.shape(), Shape{});
|
||||
CHECK_EQ(out.size(), 1);
|
||||
CHECK_EQ(out.dtype(), float32);
|
||||
CHECK_EQ(out.item<float>(), 1.0f);
|
||||
|
Reference in New Issue
Block a user