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

Fix overflow in large filter small channels (#2520)

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-08-20 08:03:29 -07:00
committed by GitHub
parent 512281781c
commit 25c1e03205
2 changed files with 8 additions and 1 deletions
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 ImplicitGemmConv2DParams* gemm_params;
short weight_hw;
int weight_hw;
const device T* src[n_rows];