LICENSE
README.md
pyproject.toml
setup.py
src/flashdiffusion/ROPE.py
src/flashdiffusion/__init__.py
src/flashdiffusion/cdc.py
src/flashdiffusion/dmap.py
src/flashdiffusion/eigensolver.py
src/flashdiffusion/eigensolver_gpu.py
src/flashdiffusion/kernel.py
src/flashdiffusion/kernel_cuda.py
src/flashdiffusion/kernel_cuda_sm120.py
src/flashdiffusion/kernel_jit.py
src/flashdiffusion/utils.py
src/flashdiffusion.egg-info/PKG-INFO
src/flashdiffusion.egg-info/SOURCES.txt
src/flashdiffusion.egg-info/dependency_links.txt
src/flashdiffusion.egg-info/requires.txt
src/flashdiffusion.egg-info/top_level.txt
src/flashdiffusion/csrc/flash_diffusion_sm120.cu
src/flashdiffusion/csrc/flash_diffusion_sm120X.cu
src/flashdiffusion/csrc/flash_diffusion_sm120_.cu
src/flashdiffusion/csrc/flash_diffusion_sm80.cu
src/flashdiffusion/csrc/flash_diffusion_sm80X.cu
src/flashdiffusion/csrc/flash_diffusion_sm80_.cu
tests/test_kernel.py