add mamba

2eefe3d6 · luopl · b7535e7c · 2eefe3d6 · 2eefe3d6 · 2eefe3d6
Commit 2eefe3d6 authored Sep 29, 2024 by luopl
5 changed files
--- a/mamba/tests/ops/test_selective_scan.py
+++ b/mamba/tests/ops/test_selective_scan.py
--- a/mamba/tests/ops/triton/test_layernorm_gated.py
+++ b/mamba/tests/ops/triton/test_layernorm_gated.py
--- a/mamba/tests/ops/triton/test_selective_state_update.py
+++ b/mamba/tests/ops/triton/test_selective_state_update.py
--- a/mamba/tests/ops/triton/test_ssd.py
+++ b/mamba/tests/ops/triton/test_ssd.py
--- a/mamba/tests/test_generation.py
+++ b/mamba/tests/test_generation.py