RWKV-CUDA
The CUDA version of the RWKV language model ( https://github.com/BlinkDL/RWKV-LM )
파일 탐색기
최종 버전 다운로드 (.zip)- timex_cuda_v0.cu
- timex_cuda_v1.cu
- timex_cuda_v2.cu
- timex_cuda_v3.cu
- timex_op.cpp
- run.py
- rwkv7_clampw.cpp
- rwkv7_clampw.cu
- rwkv7_state_clampw.cpp
- rwkv7_state_clampw.cu
- rwkv7_statepassing_clampw.cpp
- rwkv7_statepassing_clampw.cu
- rwkv7_cuda_benchmark.py
- rwkv7_cuda_benchmark_state.py
- rwkv7_cuda_benchmark_state_passing.py
- wkv_cuda_v0.cu
- wkv_cuda_v1.cu
- wkv_cuda_v2.cu
- wkv_op.cpp
- run.py
- wkv5_cuda_ref.cu
- wkv5_cuda_v1.cu
- wkv5_cuda_v1a.cu
- wkv5_cuda_v1b.cu
- wkv5_cuda_v1c.cu
- wkv5_cuda_v1d.cu
- wkv5_cuda_v1e.cu
- wkv5_cuda_v2.cu
- wkv5_cuda_v3.cu
- wkv5_op.cpp
- wkv5_ref.cpp
- run.py
- wkv5_cuda_v1.cu
- wkv5_cuda_v1a.cu
- wkv5_cuda_v1b.cu
- wkv5_cuda_v1b2.cu
- wkv5_cuda_v2.cu
- wkv5_cuda_v3.cu
- wkv5_op.cpp
- run.py
- wkv5a_cuda_v1.cu
- wkv5a_cuda_v1a.cu
- wkv5a_cuda_v1a2.cu
- wkv5a_op.cpp
- run.py
- wkv5_cuda_v1b2.cu
- wkv5_op.cpp
- wkv6_cuda_v1.cu
- wkv6_cuda_v1a.cu
- wkv6_op.cpp
- run.py
- wkv6state_cuda_v1a.cu
- wkv6state_op.cpp
- run.py
- .gitignore
- README.md
// repository documentation
Was this content helpful?
(0 ratings)