|
att_one_v5.cu
|
upgrade cuda-beta
|
2023-09-15 16:30:11 +08:00 |
|
att_one.cu
|
upgrade cuda-beta
|
2023-09-15 16:30:11 +08:00 |
|
att_seq.cu
|
upgrade cuda-beta
|
2023-09-15 16:30:11 +08:00 |
|
element_wise.h
|
add rwkv-cuda-beta support (faster)
|
2023-08-14 22:07:15 +08:00 |
|
ffn.cu
|
add rwkv-cuda-beta support (faster)
|
2023-08-14 22:07:15 +08:00 |
|
gemm_fp16_cublas.cpp
|
upgrade cuda-beta
|
2023-09-15 16:30:11 +08:00 |
|
operators.cu
|
add rwkv-cuda-beta support (faster)
|
2023-08-14 22:07:15 +08:00 |
|
util.h
|
add rwkv-cuda-beta support (faster)
|
2023-08-14 22:07:15 +08:00 |
|
wrapper.cpp
|
upgrade cuda-beta
|
2023-09-15 16:30:11 +08:00 |
