transformers
b50ff599 - [`Mamba2`] Move dt calculations to kernel (#33520)

Commit

1 year ago

[`Mamba2`] Move dt calculations to kernel (#33520) * use kernel for dt calculations * add small test * [run-slow] mamba2

References

Author

vasqu

Parents