From 1d200d53ab43be339d51069390e673a613a75ed4 Mon Sep 17 00:00:00 2001
From: josc146 <josStorer@outlook.com>
Date: Wed, 25 Oct 2023 17:51:13 +0800
Subject: [PATCH] fix beta linux kernel

---
 backend-python/rwkv_pip/beta/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend-python/rwkv_pip/beta/model.py b/backend-python/rwkv_pip/beta/model.py
index ca32e1a..1c90fbf 100644
--- 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",