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

spelling: additional

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

2
docs/src/dev/extensions.rst
View File

@ -291,7 +291,7 @@ pointwise. This is captured in the templated function :meth:`axpby_impl`.
auto y_offset = elem_to_loc(out_idx, y.shape(), y.strides()); auto y_offset = elem_to_loc(out_idx, y.shape(), y.strides());
// We allocate the output to be contiguous and regularly strided // We allocate the output to be contiguous and regularly strided
// (defaults to row major) and hence it doesn't need additonal mapping // (defaults to row major) and hence it doesn't need additional mapping
out_ptr[out_idx] = alpha * x_ptr[x_offset] + beta * y_ptr[y_offset]; out_ptr[out_idx] = alpha * x_ptr[x_offset] + beta * y_ptr[y_offset];
} }
} }

2
examples/extensions/axpby/axpby.cpp
View File

@ -98,7 +98,7 @@ void axpby_impl(
auto y_offset = elem_to_loc(out_idx, y.shape(), y.strides()); auto y_offset = elem_to_loc(out_idx, y.shape(), y.strides());
// We allocate the output to be contiguous and regularly strided // We allocate the output to be contiguous and regularly strided
// (defaults to row major) and hence it doesn't need additonal mapping // (defaults to row major) and hence it doesn't need additional mapping
out_ptr[out_idx] = alpha * x_ptr[x_offset] + beta * y_ptr[y_offset]; out_ptr[out_idx] = alpha * x_ptr[x_offset] + beta * y_ptr[y_offset];
} }
} }