From bd0622c4d9ca2601338064b5e6de6de8bb75ce95 Mon Sep 17 00:00:00 2001
From: Angelos Katharopoulos <a_katharopoulos@apple.com>
Date: Fri, 4 Jul 2025 13:16:54 -0700
Subject: [PATCH] Address floating point exception on linux blas

---
 mlx/backend/cpu/masked_mm.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mlx/backend/cpu/masked_mm.cpp b/mlx/backend/cpu/masked_mm.cpp
index cd0680131..fb277b530 100644
--- a/mlx/backend/cpu/masked_mm.cpp
+++ b/mlx/backend/cpu/masked_mm.cpp
@@ -80,6 +80,9 @@ inline void segmented_mm(
         segments[elem_to_loc(2 * i, segments_shape, segments_strides)];
     uint32_t k_end =
         segments[elem_to_loc(2 * i + 1, segments_shape, segments_strides)];
+    if (k_end <= k_start) {
+      continue;
+    }
     a_copy[ndim - 1] = k_end - k_start;
     b_copy[ndim - 2] = k_end - k_start;
     matmul<T>(