LICENSE
MANIFEST.in
NOTICE
README.md
pyproject.toml
setup.py
FlashRWKV2.egg-info/PKG-INFO
FlashRWKV2.egg-info/SOURCES.txt
FlashRWKV2.egg-info/dependency_links.txt
FlashRWKV2.egg-info/not-zip-safe
FlashRWKV2.egg-info/requires.txt
FlashRWKV2.egg-info/top_level.txt
LICENSES/Apache-2.0.txt
csrc/bindings.cpp
csrc/bindings.h
csrc/registration.cpp
csrc/validation.cpp
csrc/validation.h
csrc/sm80/sampling/infer_fp32_forward_varlen.cpp
csrc/sm80/sampling/infer_fp32_forward_varlen.cu
csrc/validation/recurrent_metadata.cpp
csrc/validation/recurrent_metadata.cu
flashrwkv2/__init__.py
flashrwkv2/compile.py
flashrwkv2/_internal/elementwise.py
flashrwkv2/_internal/linear.py
flashrwkv2/_internal/logprob.py
flashrwkv2/_internal/state.py
flashrwkv2/_internal/tuning.py
flashrwkv2/cmix/__init__.py
flashrwkv2/embedding/__init__.py
flashrwkv2/head/l2wrap_ce/__init__.py
flashrwkv2/head/linear/__init__.py
flashrwkv2/loss/l2wrap_ce/__init__.py
flashrwkv2/loss/l2wrap_ce/tilelang.py
flashrwkv2/post_norm/__init__.py
flashrwkv2/post_norm/tilelang.py
flashrwkv2/sampling/__init__.py
flashrwkv2/tmix/a_gate/__init__.py
flashrwkv2/tmix/kk_pre/__init__.py
flashrwkv2/tmix/kk_pre/tilelang.py
flashrwkv2/tmix/readout/__init__.py
flashrwkv2/tmix/readout/pretrain.py
flashrwkv2/tmix/readout/tilelang.py
flashrwkv2/tmix/tokenshift/__init__.py
flashrwkv2/tmix/tokenshift/tilelang.py
flashrwkv2/tmix/vres_gate/__init__.py
flashrwkv2/tmix/vres_gate/tilelang.py
flashrwkv2/tmix/wkv7/__init__.py
flashrwkv2/tmix/wkv7/pretrain.py
flashrwkv2/tmix/wkv7/statetune.py
flashrwkv2/tmix/wkv7/tilelang.py
flashrwkv2/tmix/wkv_prepare/__init__.py
flashrwkv2/tmix/wkv_prepare/tilelang.py
tests/test_runtime.py