fix beta linux kernel

2023-10-25 17:51:13 +08:00 · 2023-10-25 17:51:13 +08:00 · 1d200d53ab
commit 1d200d53ab
parent df9e1f408e
1 changed files with 1 additions and 1 deletions
--- a/backend-python/rwkv_pip/beta/model.py
+++ b/backend-python/rwkv_pip/beta/model.py
@ -94,7 +94,7 @@ if os.environ.get("RWKV_CUDA_ON") == "1":
                f"{current_path}/cuda/att_one_v5.cu",
            ],
            verbose=True,
-            extra_ldflags=["cublas.lib"],
+            extra_ldflags=["cublas.lib" if os.name == "nt" else ""],
            extra_cuda_cflags=[
                "-t 4",
                "-std=c++17",