upgrade cuda-beta

2023-09-15 16:30:11 +08:00
parent c4042bbfd8
commit df969fcfc6
6 changed files with 601 additions and 90 deletions
--- a/backend-python/rwkv_pip/beta/cuda/att_seq.cu
+++ b/backend-python/rwkv_pip/beta/cuda/att_seq.cu
@@ -8,7 +8,6 @@

 using torch::Tensor;

-void gemm_fp16_cublas(Tensor a, Tensor b, Tensor c);
 void gemm_fp16_cublas(const void *a, const void *b, void *c, int m,
                      int n, int k, bool output_fp32);