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

Use int64 stride everywhere (#1671)

Browse Source 
* use int64 stride everywhere

* fix ext

* fix ext

* more shape + cleanup

* one more

* few more
This commit is contained in:
Awni Hannun
2024-12-09 11:09:02 -08:00
committed by GitHub
parent 35b412c099
commit 40c62c1321
102 changed files with 1262 additions and 1705 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/arg_reduce_tests.cpp
View File

@@ -11,7 +11,7 @@ void test_arg_reduce_small(
Device d,
const array& x,
ArgReduce::ReduceType r,
std::vector<int> out_shape,
Shape out_shape,
int axis,
std::vector<int> expected_output) {
auto s = default_stream(d);
@@ -27,7 +27,7 @@ void test_arg_reduce_small(
void test_arg_reduce_against_cpu(
const array& x,
ArgReduce::ReduceType r,
std::vector<int> out_shape,
Shape out_shape,
int axis) {
auto y1 = array(
out_shape,
@@ -125,7 +125,7 @@ TEST_CASE("test arg reduce against cpu") {
void test_arg_reduce_small_bool(
Device d,
ArgReduce::ReduceType r,
std::vector<int> out_shape,
Shape out_shape,
int axis,
std::vector<int> expected_output) {
auto s = default_stream(d);