From ec13e549d3e1de13d05af759cc8bef3f7cf5e318 Mon Sep 17 00:00:00 2001 From: Asaf Gardin <39553475+Josephasafg@users.noreply.github.com> Date: Thu, 26 Feb 2026 14:22:06 +0200 Subject: [PATCH] [Bugfix] Fix uint32 overflow in Mamba selective scan state pointer arithmetic (#35275) Signed-off-by: Josephasafg --- csrc/mamba/mamba_ssm/selective_scan.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/csrc/mamba/mamba_ssm/selective_scan.h b/csrc/mamba/mamba_ssm/selective_scan.h index 7d22dd8b8..e93455a57 100644 --- a/csrc/mamba/mamba_ssm/selective_scan.h +++ b/csrc/mamba/mamba_ssm/selective_scan.h @@ -15,7 +15,7 @@ //////////////////////////////////////////////////////////////////////////////////////////////////// struct SSMParamsBase { - using index_t = uint32_t; + using index_t = size_t; int batch, dim, seqlen, dstate, n_groups, n_chunks; int dim_ngroups_ratio;