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

Fix the input loading for small channels large filters

Browse Source
This commit is contained in:
Angelos Katharopoulos 2025-08-19 23:41:48 -07:00
parent cea9369610
commit 2581a9ab85
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h
View File

@ -83,7 +83,7 @@ struct Conv2DInputBlockLoaderSmallChannels {
const constant MLXConvParams<2>* params; const constant MLXConvParams<2>* params;
const constant ImplicitGemmConv2DParams* gemm_params; const constant ImplicitGemmConv2DParams* gemm_params;
short weight_hw; int weight_hw;
const device T* src[n_rows]; const device T* src[n_rows];