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/sm120/cmix/mix/infer_fp16_forward_varlen.cpp
csrc/sm120/cmix/mix/infer_fp16_forward_varlen.cu
csrc/sm120/cmix/sparse/infer_fp16_forward_varlen.cpp
csrc/sm120/cmix/sparse/infer_fp16_forward_varlen.cu
csrc/sm120/embedding/infer_fp16_forward_varlen.cpp
csrc/sm120/embedding/infer_fp16_forward_varlen.cu
csrc/sm120/head/linear/infer_fp16_forward_varlen.cpp
csrc/sm120/head/linear/infer_fp16_forward_varlen.cu
csrc/sm120/sampling/infer_fp32_forward_varlen.cpp
csrc/sm120/sampling/infer_fp32_forward_varlen.cu
csrc/sm120/tmix/kk_a_gate/infer_fp16_forward_varlen.cpp
csrc/sm120/tmix/kk_a_gate/infer_fp16_forward_varlen.cu
csrc/sm120/tmix/linear/infer_fp16_forward_varlen.cpp
csrc/sm120/tmix/linear/infer_fp16_forward_varlen.cu
csrc/sm120/tmix/lnx_rkvres_xg/infer_fp16_forward_varlen.cpp
csrc/sm120/tmix/lnx_rkvres_xg/infer_fp16_forward_varlen.cu
csrc/sm120/tmix/mix6/infer_fp16_forward_varlen.cpp
csrc/sm120/tmix/mix6/infer_fp16_forward_varlen.cu
csrc/sm120/tmix/normalization/infer_fp16_forward_varlen.cpp
csrc/sm120/tmix/normalization/infer_fp16_forward_varlen.cu
csrc/sm120/tmix/vres_gate/infer_fp16_forward_varlen.cpp
csrc/sm120/tmix/vres_gate/infer_fp16_forward_varlen.cu
csrc/sm120/tmix/wkv7/infer_chunk_bf16_forward_varlen.cpp
csrc/sm120/tmix/wkv7/infer_chunk_bf16_forward_varlen.cu
csrc/sm120/tmix/wkv7/infer_recurrent_fp16_forward_varlen.cpp
csrc/sm120/tmix/wkv7/infer_recurrent_fp16_forward_varlen.cu
csrc/sm120/tmix/wkv7/infer_recurrent_fp32io16_forward_varlen.cpp
csrc/sm120/tmix/wkv7/infer_recurrent_fp32io16_forward_varlen.cu
csrc/sm120/tmix/wkv7/recurrent_decay.cuh
csrc/sm90/cmix/mix/pretrain_bf16_backward.cpp
csrc/sm90/cmix/mix/pretrain_bf16_backward.cu
csrc/sm90/cmix/mix/pretrain_bf16_forward.cpp
csrc/sm90/cmix/mix/pretrain_bf16_forward.cu
csrc/sm90/cmix/mix/statetune_bf16_backward.cpp
csrc/sm90/cmix/mix/statetune_bf16_backward.cu
csrc/sm90/cmix/mix/statetune_bf16_forward.cpp
csrc/sm90/cmix/mix/statetune_bf16_forward.cu
csrc/sm90/head/l2wrap_ce/pretrain_bf16_forward.cpp
csrc/sm90/head/l2wrap_ce/pretrain_bf16_forward.cu
csrc/sm90/loss/l2wrap_ce/pretrain_bf16_backward.cpp
csrc/sm90/loss/l2wrap_ce/pretrain_bf16_backward.cu
csrc/sm90/loss/l2wrap_ce/pretrain_bf16_forward.cpp
csrc/sm90/loss/l2wrap_ce/pretrain_bf16_forward.cu
csrc/sm90/rl_infctx/wkv7/rl_infctx_chunk_fp32io16_backward.cpp
csrc/sm90/rl_infctx/wkv7/rl_infctx_chunk_fp32io16_backward.cu
csrc/sm90/rl_infctx/wkv7/rl_infctx_chunk_fp32io16_forward.cpp
csrc/sm90/rl_infctx/wkv7/rl_infctx_chunk_fp32io16_forward.cu
csrc/sm90/rl_infctx/wkv7/rl_infctx_chunk_fp32io16_replay.cuh
csrc/sm90/tmix/a_gate/pretrain_bf16_backward.cpp
csrc/sm90/tmix/a_gate/pretrain_bf16_backward.cu
csrc/sm90/tmix/a_gate/pretrain_bf16_forward.cpp
csrc/sm90/tmix/a_gate/pretrain_bf16_forward.cu
csrc/sm90/tmix/kk_pre/pretrain_bf16_backward.cpp
csrc/sm90/tmix/kk_pre/pretrain_bf16_backward.cu
csrc/sm90/tmix/kk_pre/pretrain_bf16_forward.cpp
csrc/sm90/tmix/kk_pre/pretrain_bf16_forward.cu
csrc/sm90/tmix/lnx_rkvres_xg/pretrain_bf16_backward.cpp
csrc/sm90/tmix/lnx_rkvres_xg/pretrain_bf16_backward.cu
csrc/sm90/tmix/lnx_rkvres_xg/pretrain_bf16_forward.cpp
csrc/sm90/tmix/lnx_rkvres_xg/pretrain_bf16_forward.cu
csrc/sm90/tmix/mix6/pretrain_bf16_backward.cpp
csrc/sm90/tmix/mix6/pretrain_bf16_backward.cu
csrc/sm90/tmix/mix6/pretrain_bf16_forward.cpp
csrc/sm90/tmix/mix6/pretrain_bf16_forward.cu
csrc/sm90/tmix/mix6/statetune_bf16_backward.cpp
csrc/sm90/tmix/mix6/statetune_bf16_backward.cu
csrc/sm90/tmix/mix6/statetune_bf16_forward.cpp
csrc/sm90/tmix/mix6/statetune_bf16_forward.cu
csrc/sm90/tmix/vres_gate/pretrain_bf16_backward.cpp
csrc/sm90/tmix/vres_gate/pretrain_bf16_backward.cu
csrc/sm90/tmix/vres_gate/pretrain_bf16_forward.cpp
csrc/sm90/tmix/vres_gate/pretrain_bf16_forward.cu
csrc/sm90/tmix/wkv7/pretrain_recurrent_bf16_forward.cpp
csrc/sm90/tmix/wkv7/pretrain_recurrent_bf16_forward.cu
csrc/sm90/tmix/wkv7/statetune_recurrent_fp32io16_backward.cpp
csrc/sm90/tmix/wkv7/statetune_recurrent_fp32io16_backward.cu
csrc/sm90/tmix/wkv7/statetune_recurrent_fp32io16_forward.cpp
csrc/sm90/tmix/wkv7/statetune_recurrent_fp32io16_forward.cu
csrc/validation/recurrent_metadata.cu
flashrwkv2/__init__.py
flashrwkv2/cmix/mix/__init__.py
flashrwkv2/cmix/sparse/__init__.py
flashrwkv2/embedding/__init__.py
flashrwkv2/head/l2wrap_ce/__init__.py
flashrwkv2/head/linear/__init__.py
flashrwkv2/loss/l2wrap_ce/__init__.py
flashrwkv2/rl_infctx/__init__.py
flashrwkv2/rl_infctx/wkv7/__init__.py
flashrwkv2/sampling/__init__.py
flashrwkv2/tmix/a_gate/__init__.py
flashrwkv2/tmix/kk_a_gate/__init__.py
flashrwkv2/tmix/kk_pre/__init__.py
flashrwkv2/tmix/linear/__init__.py
flashrwkv2/tmix/lnx_rkvres_xg/__init__.py
flashrwkv2/tmix/lnx_rkvres_xg/pretrain.py
flashrwkv2/tmix/mix6/__init__.py
flashrwkv2/tmix/normalization/__init__.py
flashrwkv2/tmix/vres_gate/__init__.py
flashrwkv2/tmix/wkv7/__init__.py
flashrwkv2/tmix/wkv7/chunk.py
flashrwkv2/tmix/wkv7/pretrain.py
flashrwkv2/tmix/wkv7/statetune.py