There is still room for improvement, but it works! * cuda : adapt Mamba1 ssm scan to shape changes from Mamba2 |
||
|---|---|---|
| .. | ||
| cmake | ||
| include | ||
| src | ||
| .gitignore | ||
| CMakeLists.txt | ||
There is still room for improvement, but it works! * cuda : adapt Mamba1 ssm scan to shape changes from Mamba2 |
||
|---|---|---|
| .. | ||
| cmake | ||
| include | ||
| src | ||
| .gitignore | ||
| CMakeLists.txt | ||