From 7391cfab9ebbd2015d7afdc914695ac9a1f3c088 Mon Sep 17 00:00:00 2001
From: Clemens Giuliani <clemens@inailuig.it>
Date: Mon, 3 Nov 2025 17:43:36 +0100
Subject: [PATCH] [CUDA] fix bug in syevBatched ffi kernel

---
 jaxlib/gpu/solver_kernels_ffi.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/jaxlib/gpu/solver_kernels_ffi.cc b/jaxlib/gpu/solver_kernels_ffi.cc
index 2ac4a72de99e..b25294c641a7 100644
--- a/jaxlib/gpu/solver_kernels_ffi.cc
+++ b/jaxlib/gpu/solver_kernels_ffi.cc
@@ -539,7 +539,7 @@ ffi::Error Syevd64Impl(int64_t batch, int64_t n, gpuStream_t stream,
                        IsSyevBatchedSupported());
   if (is_batched_syev_supported) {
     int64_t matrix_size = n * n * ffi::ByteWidth(dataType);
-    batch_step = std::numeric_limits<int>::max() / matrix_size;
+    batch_step = std::max(int64_t(1), std::numeric_limits<int>::max() / matrix_size);
     if (batch_step >= 32 * 1024) {
       batch_step = 32 * 1024;
     }