From 10374c298d0953d4c0fb8799b9cab6561291f9e0 Mon Sep 17 00:00:00 2001 From: petrukha-ivan Date: Tue, 13 Oct 2026 16:19:54 +0200 Subject: [PATCH] Fix missing stream in gather matmul indices cast --- mlx/ops.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlx/ops.cpp b/mlx/ops.cpp index b58a2836e4..8299320198 100644 --- a/mlx/ops.cpp +++ b/mlx/ops.cpp @@ -96,7 +96,7 @@ array indices_or_default( << " Got indices with invalid dtype. Indices must be integral."; throw std::invalid_argument(msg.str()); } - return astype(indices.value(), uint32); + return astype(indices.value(), uint32, s); } Shape shape(x.shape().begin(), x.shape().end() - 2);