diff --git a/mlx/backend/metal/utils.h b/mlx/backend/metal/utils.h
index 1a49b4f51..f9245a6d6 100644
--- a/mlx/backend/metal/utils.h
+++ b/mlx/backend/metal/utils.h
@@ -88,7 +88,7 @@ inline int get_work_per_thread(Dtype dtype) {
   return std::max(1, 8 / dtype.size());
 }
 
-inline int ceildiv(int n, int m) {
+inline size_t ceildiv(size_t n, size_t m) {
   return (n + m - 1) / m;
 }