L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Metal] Fix 32-bit integer overflow in conv3d unfold kernel (#3143)

Browse Source
This commit is contained in:
Kellen Sun
2026-02-19 10:01:22 -08:00
committed by GitHub
parent f2f2d16451
commit d4c81062ad
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mlx/backend/metal/kernels/conv.metal
+3 -2
View File
@@ -30,7 +30,7 @@ template <typename T, int N>
out_pixels *= params->oS[i];
// Set out
out += gid.z * filter_size + gid.y * (params->C);
out += (size_t)gid.z * filter_size + (size_t)gid.y * (params->C);
// Coordinates in input
int is[N] = {0};
@@ -93,7 +93,8 @@ template <typename T, int N>
out_pixels *= params->oS[i];
// Set out
out += gid.z * filter_size + gid.x * (filter_size / params->C);
out +=
(size_t)gid.z * filter_size + (size_t)gid.x * (filter_size / params->C);
// Coordinates in input
int is[N] = {0};