修改readme

25991f98 · hepj · ac192496 · 25991f98 · 25991f98 · 25991f98
Commit 25991f98 authored Jul 25, 2024 by hepj
13 changed files
--- a/mamba_ssm/ops/triton/ssd_chunk_scan.py
+++ b/mamba_ssm/ops/triton/ssd_chunk_scan.py
--- a/mamba_ssm/ops/triton/ssd_chunk_state.py
+++ b/mamba_ssm/ops/triton/ssd_chunk_state.py
--- a/mamba_ssm/ops/triton/ssd_combined.py
+++ b/mamba_ssm/ops/triton/ssd_combined.py
--- a/mamba_ssm/ops/triton/ssd_state_passing.py
+++ b/mamba_ssm/ops/triton/ssd_state_passing.py
--- a/mamba_ssm/utils/__init__.py
+++ b/mamba_ssm/utils/__init__.py
--- a/mamba_ssm/utils/generation.py
+++ b/mamba_ssm/utils/generation.py
--- a/mamba_ssm/utils/hf.py
+++ b/mamba_ssm/utils/hf.py
--- a/rocm_patch/rocm6_0.patch
+++ b/rocm_patch/rocm6_0.patch
--- a/tests/ops/test_selective_scan.py
+++ b/tests/ops/test_selective_scan.py
--- a/tests/ops/triton/test_layernorm_gated.py
+++ b/tests/ops/triton/test_layernorm_gated.py
--- a/tests/ops/triton/test_selective_state_update.py
+++ b/tests/ops/triton/test_selective_state_update.py
--- a/tests/ops/triton/test_ssd.py
+++ b/tests/ops/triton/test_ssd.py
--- a/tests/test_generation.py
+++ b/tests/test_generation.py