diff --git a/mlx/backend/metal/kernels/conv.metal b/mlx/backend/metal/kernels/conv.metal index 925fa7f69..be6fd6d1e 100644 --- a/mlx/backend/metal/kernels/conv.metal +++ b/mlx/backend/metal/kernels/conv.metal @@ -773,9 +773,7 @@ template < const int o_idx = BO * tid.x + thr_t; // Divide tile into h, w tile - uniform oHu = make_uniform(params.oS[0]); uniform oWu = make_uniform(params.oS[1]); - uniform tHu = (oHu + make_uniform(FN - 1)) / make_uniform(FN); uniform tWu = (oWu + make_uniform(FN - 1)) / make_uniform(FN); const int oH_idx = FN * (t_idx / tWu); @@ -1056,4 +1054,4 @@ template < instantiate_winograd_conv_2d_fused_2(float32, float); instantiate_winograd_conv_2d_fused_2(float16, float16_t); instantiate_winograd_conv_2d_fused_2(bfloat16, bfloat16_t); -// clang-format on \ No newline at end of file +// clang-format on