zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-21 20:46:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

spelling: buffer

Browse Source 
Signed-off-by: Josh Soref <2119212+jsoref@users.noreply.github.com>
This commit is contained in:
Josh Soref 2024-01-01 22:50:55 -05:00
parent 16046082f6
commit edc830dcd3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/backend/metal/indexing.cpp
View File

@ -102,7 +102,7 @@ void Gather::eval_gpu(const std::vector<array>& inputs, array& out) {
static_cast<size_t*>(idx_strides_buf.raw_ptr()) + i * idx_ndim); static_cast<size_t*>(idx_strides_buf.raw_ptr()) + i * idx_ndim);
} }
// Allocate the argument bufer // Allocate the argument buffer
auto arg_buf = allocator::malloc_or_wait(arg_enc->encodedLength()); auto arg_buf = allocator::malloc_or_wait(arg_enc->encodedLength());
// Register data with the encoder // Register data with the encoder
@ -246,7 +246,7 @@ void Scatter::eval_gpu(const std::vector<array>& inputs, array& out) {
static_cast<size_t*>(idx_strides_buf.raw_ptr()) + i * idx_ndim); static_cast<size_t*>(idx_strides_buf.raw_ptr()) + i * idx_ndim);
} }
// Allocate the argument bufer // Allocate the argument buffer
auto arg_buf = allocator::malloc_or_wait(arg_enc->encodedLength()); auto arg_buf = allocator::malloc_or_wait(arg_enc->encodedLength());
// Register data with the encoder // Register data with the encoder