From 0005816c1d0432444c51421ecc61fc51ddfa3843 Mon Sep 17 00:00:00 2001
From: josc146 <josStorer@outlook.com>
Date: Thu, 5 Oct 2023 00:08:18 +0800
Subject: [PATCH] fix linux kernel (partial revert 68228a45)

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

diff --git a/backend-python/rwkv_pip/model.py b/backend-python/rwkv_pip/model.py
index c92f94d..81469b9 100644
--- a/backend-python/rwkv_pip/model.py
+++ b/backend-python/rwkv_pip/model.py
@@ -92,7 +92,7 @@ if os.environ.get("RWKV_CUDA_ON") == "1":
                     f"{current_path}/cuda/gemm_fp16_cublas.cpp",
                 ],
                 verbose=True,
-                extra_ldflags=["cublas.lib"],
+                extra_ldflags=["cublas.lib" if os.name == "nt" else ""],
                 extra_cuda_cflags=[
                     "--use_fast_math",
                     "-O3",
@@ -596,6 +596,7 @@ class RWKV(MyModule):
                             "-res-usage",
                             "--use_fast_math",
                             "-O3",
+                            "-Xptxas -O3" if os.name != "nt" else "",
                             "--extra-device-vectorization",
                             f"-D_N_={HEAD_SIZE}",
                         ],