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

fix mismatch

Browse Source
This commit is contained in:
tianyi 2025-07-22 18:00:42 +08:00
parent b2f0ebe9ee
commit 7df3a2887d
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mlx/backend/metal/quantized.cpp
View File

@ -259,10 +259,10 @@ void qmv_no_parallel_m(
const Stream& s) { const Stream& s) {
int B = out.size() / M / N; int B = out.size() / M / N;
int bn = 128; int bn = 8;
// int bk = 32; int bk = 32;
MTL::Size group_dims(2, 1, 1); MTL::Size group_dims(bk, 2, 1);
MTL::Size grid_dims((N + bn - 1) / bn, 1, B); MTL::Size grid_dims(1, (N + bn - 1) / bn, B);
std::string kname; std::string kname;
kname.reserve(64); kname.reserve(64);