PR #33520 [`Mamba2`] Move dt calculations to kernel

[`Mamba2`] Move dt calculations to kernel #33520

amyeroberts merged 3 commits into huggingface:main from vasqu:mamba2-fast-inference-path-fix

molbap approved these changes on 2024-09-17

molbap requested a review from

amyeroberts 1 year ago

amyeroberts approved these changes on 2024-09-17

amyeroberts added run-slow

use kernel for dt calculations

61e976c9

add small test

fe01458a

[run-slow] mamba2

ff0cb7cd

vasqu force pushed to ff0cb7cd 1 year ago

amyeroberts merged b50ff599 into main 1 year ago

vasqu deleted the mamba2-fast-inference-path-fix branch 1 year ago

Reviewers

molbap

amyeroberts

Assignees

No one assigned

Labels

run-slow

Milestone

No milestone