huggingface / optimum-habana

Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)

Apache License 2.0

154 stars 202 forks source link

Open zzhang37 opened 3 days ago

zzhang37 commented 3 days ago

What does this PR do?

Added custom mamba pscan op and fixed the mamba cache issue

Fixes # (issue)

[ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
[ ] Did you make sure to update the documentation with your changes?
[ ] Did you write any new necessary tests?

jiminha commented 15 hours ago

@zzhang37 Also, all synapse dependencies merged in to 1.19 release?