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: among

Browse Source 
Signed-off-by: Josh Soref <2119212+jsoref@users.noreply.github.com>
This commit is contained in:
Josh Soref 2024-01-01 22:28:02 -05:00
parent 33be1a27af
commit 7057f13b5f
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/kernels/gemv.metal
View File

@ -28,7 +28,7 @@ struct GEMVKernel {
static_assert(BN == SIMD_SIZE, "gemv block must have a width of SIMD_SIZE"); static_assert(BN == SIMD_SIZE, "gemv block must have a width of SIMD_SIZE");
// - The matrix of size (M = out_vec_size, N = in_vec_size) is divided up // - The matrix of size (M = out_vec_size, N = in_vec_size) is divided up
// into blocks of (BM * TM, BN * TN) divided amoung threadgroups // into blocks of (BM * TM, BN * TN) divided among threadgroups
// - Every thread works on a block of (TM, TN) // - Every thread works on a block of (TM, TN)
// - We assume each thead group is launched with (BN, BM, 1) threads // - We assume each thead group is launched with (BN, BM, 1) threads
// //
@ -166,7 +166,7 @@ template <
struct GEMVTKernel { struct GEMVTKernel {
// - The matrix of size (M = in_vec_size, N = out_vec_size) is divided up // - The matrix of size (M = in_vec_size, N = out_vec_size) is divided up
// into blocks of (BM * TM, BN * TN) divided amoung threadgroups // into blocks of (BM * TM, BN * TN) divided among threadgroups
// - Every thread works on a block of (TM, TN) // - Every thread works on a block of (TM, TN)
// - We assume each thead group is launched with (BN, BM, 1) threads // - We assume each thead group is launched with (BN, BM, 1) threads
// //