From e041e3967fd5d1b2c8cca3112f4823137541ecf5 Mon Sep 17 00:00:00 2001 From: Feli Date: Sun, 9 Aug 2026 02:05:32 +0800 Subject: [PATCH] Fix abort when vector elements=0 --- mlx/backend/metal/device.h | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mlx/backend/metal/device.h b/mlx/backend/metal/device.h index 871e95ccff..9b883ba8ab 100644 --- a/mlx/backend/metal/device.h +++ b/mlx/backend/metal/device.h @@ -64,8 +64,10 @@ class MLX_API CommandEncoder { template >> void set_vector_bytes(const Vec& vec, size_t nelems, int idx) { - get_command_encoder()->setBytes( - vec.data(), nelems * sizeof(typename Vec::value_type), idx); + if (nelems > 0) { + get_command_encoder()->setBytes( + vec.data(), nelems * sizeof(typename Vec::value_type), idx); + } } template >> void set_vector_bytes(const Vec& vec, int idx) {