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

Remove unused uniform (#1867)

Browse Source
This commit is contained in:
Angelos Katharopoulos 2025-02-14 15:51:41 -08:00 committed by GitHub
parent 6cec78d8f2
commit 1762793989
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/backend/metal/kernels/conv.metal
View File

@ -773,9 +773,7 @@ template <
const int o_idx = BO * tid.x + thr_t; const int o_idx = BO * tid.x + thr_t;
// Divide tile into h, w tile // Divide tile into h, w tile
uniform<int> oHu = make_uniform(params.oS[0]);
uniform<int> oWu = make_uniform(params.oS[1]); uniform<int> oWu = make_uniform(params.oS[1]);
uniform<int> tHu = (oHu + make_uniform(FN - 1)) / make_uniform(FN);
uniform<int> tWu = (oWu + make_uniform(FN - 1)) / make_uniform(FN); uniform<int> tWu = (oWu + make_uniform(FN - 1)) / make_uniform(FN);
const int oH_idx = FN * (t_idx / tWu); 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(float32, float);
instantiate_winograd_conv_2d_fused_2(float16, float16_t); instantiate_winograd_conv_2d_fused_2(float16, float16_t);
instantiate_winograd_conv_2d_fused_2(bfloat16, bfloat16_t); instantiate_winograd_conv_2d_fused_2(bfloat16, bfloat16_t);
// clang-format on // clang-format on