From 2581a9ab85e793ac59e04f171bf18ab0e4b9c050 Mon Sep 17 00:00:00 2001 From: Angelos Katharopoulos Date: Tue, 19 Aug 2025 23:41:48 -0700 Subject: [PATCH] Fix the input loading for small channels large filters --- mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h b/mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h index b0b98d21a..2312e1ca6 100644 --- a/mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h +++ b/mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h @@ -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];