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

Add contiguous_copy_cpu util for copying array (#2397)

Browse Source
This commit is contained in:
Cheng
2025-07-21 23:30:35 +09:00
committed by GitHub
parent f55c4ed1d6
commit 56cc858af9
8 changed files with 17 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/backend/cpu/masked_mm.cpp
View File

@@ -136,9 +136,8 @@ void BlockMaskedMM::eval_cpu(const std::vector<array>& inputs, array& out) {
}
return std::make_tuple(true, sty, arr, false);
} else {
array arr_copy(arr.shape(), arr.dtype(), nullptr, {});
copy_cpu(arr, arr_copy, CopyType::General, s);
int64_t stx = arr.shape(-1);
array arr_copy = contiguous_copy_cpu(arr, s);
return std::make_tuple(false, stx, arr_copy, true);
}
};