Hi, I wonder the FLOps of prim::PythonOp.MambaInnerFn, could you please provide the code ? - Githubissues

state-spaces / mamba

Mamba SSM architecture

Apache License 2.0

13.2k stars 1.12k forks source link

Hi, I wonder the FLOps of prim::PythonOp.MambaInnerFn, could you please provide the code ? #559

Open 924973292 opened 2 months ago