├── README.en.md ├── README.md ├── figures └── prefix-sum.png └── mamba └── csrc └── selective_scan ├── reverse_scan.cuh ├── selective_scan.cpp ├── selective_scan.h ├── selective_scan_bwd_bf16_complex.cu ├── selective_scan_bwd_bf16_real.cu ├── selective_scan_bwd_fp16_complex.cu ├── selective_scan_bwd_fp16_real.cu ├── selective_scan_bwd_fp32_complex.cu ├── selective_scan_bwd_fp32_real.cu ├── selective_scan_bwd_kernel.cuh ├── selective_scan_common.h ├── selective_scan_fwd_bf16.cu ├── selective_scan_fwd_fp16.cu ├── selective_scan_fwd_fp32.cu ├── selective_scan_fwd_kernel.cuh ├── static_switch.h └── uninitialized_copy.cuh /README.en.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/README.en.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/README.md -------------------------------------------------------------------------------- /figures/prefix-sum.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/figures/prefix-sum.png -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/reverse_scan.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/reverse_scan.cuh -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan.cpp -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan.h -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_bwd_bf16_complex.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_bwd_bf16_complex.cu -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_bwd_bf16_real.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_bwd_bf16_real.cu -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_bwd_fp16_complex.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_bwd_fp16_complex.cu -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_bwd_fp16_real.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_bwd_fp16_real.cu -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_bwd_fp32_complex.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_bwd_fp32_complex.cu -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_bwd_fp32_real.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_bwd_fp32_real.cu -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_bwd_kernel.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_bwd_kernel.cuh -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_common.h -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_fwd_bf16.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_fwd_bf16.cu -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_fwd_fp16.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_fwd_fp16.cu -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_fwd_fp32.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_fwd_fp32.cu -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/selective_scan_fwd_kernel.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/selective_scan_fwd_kernel.cuh -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/static_switch.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/static_switch.h -------------------------------------------------------------------------------- /mamba/csrc/selective_scan/uninitialized_copy.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YouHuang67/mamba-code-explained/HEAD/mamba/csrc/selective_scan/uninitialized_copy.cuh --------------------------------------------------------------------------------