GitHub - sophiawisdom/ssms: GPU kernels for state space models

sophiawisdom / ssms Public

GPU kernels for state space models

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 27 Commits
.gitignore		.gitignore
flash_blocksparse_triton_attention.py		flash_blocksparse_triton_attention.py
install.sh		install.sh
matmul.py		matmul.py
mimo.cpp		mimo.cpp
mimo.cu		mimo.cu
mimo_done.ptx		mimo_done.ptx
monarch.cpp		monarch.cpp
monarch.cu		monarch.cu
monarch.ptx		monarch.ptx
monarch.py		monarch.py
monarch_from_matmul.ptx		monarch_from_matmul.ptx
monarch_triton.py		monarch_triton.py
run.py		run.py
run_mimo.py		run_mimo.py
run_mimo_kernel.py		run_mimo_kernel.py
run_nonbatched.py		run_nonbatched.py
siso.cpp		siso.cpp
siso.cu		siso.cu
siso_backward.ptx		siso_backward.ptx
siso_forward.ptx		siso_forward.ptx
triton_matmul.ptx		triton_matmul.ptx
triton_matmul_2.ptx		triton_matmul_2.ptx