theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix beta linux kernel

Browse Source
This commit is contained in:
josc146 2023-10-25 17:51:13 +08:00
parent df9e1f408e
commit 1d200d53ab
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend-python/rwkv_pip/beta/model.py vendored
View File

@ -94,7 +94,7 @@ if os.environ.get("RWKV_CUDA_ON") == "1":
f"{current_path}/cuda/att_one_v5.cu", f"{current_path}/cuda/att_one_v5.cu",
], ],
verbose=True, verbose=True,
extra_ldflags=["cublas.lib"], extra_ldflags=["cublas.lib" if os.name == "nt" else ""],
extra_cuda_cflags=[ extra_cuda_cflags=[
"-t 4", "-t 4",
"-std=c++17", "-std=c++17",