Skip to content

GitLab

Unverified Commit ec13e549 authored by Asaf Gardin's avatar Asaf Gardin Committed by GitHub
Browse files

[Bugfix] Fix uint32 overflow in Mamba selective scan state pointer arithmetic (#35275) 


Signed-off-by: default avatarJosephasafg <ajgard7@gmail.com>
parent c6ca5159
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
csrc/mamba/mamba_ssm/selective_scan.h
View file @ ec13e549
......@@ -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;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment