From 5bceb408bc62d1da106e7b48fef28fe739404722 Mon Sep 17 00:00:00 2001
From: liym27 <33742067+liym27@users.noreply.github.com>
Date: Wed, 3 Sep 2025 20:57:33 +0800
Subject: [PATCH 1/2] fix bug

---
 ops/csrc/setup.py | 28 +++++-----------------------
 1 file changed, 5 insertions(+), 23 deletions(-)

diff --git a/ops/csrc/setup.py b/ops/csrc/setup.py
index 548a467f8fcb..be0c9584a732 100644
--- a/ops/csrc/setup.py
+++ b/ops/csrc/setup.py
@@ -53,20 +53,6 @@ def run_single(func):
     p.join()
 
 
-def run_multi(func_list):
-    processes = []
-    for func in func_list:
-        processes.append(multiprocessing.Process(target=func))
-        processes.append(multiprocessing.Process(target=func))
-        processes.append(multiprocessing.Process(target=func))
-
-    for p in processes:
-        p.start()
-
-    for p in processes:
-        p.join()
-
-
 cc_flag = get_gencode_flags(compiled_all=False)
 cc = get_sm_version()
 
@@ -256,12 +242,8 @@ def setup_paddle_bwd_ops():
 
 
 if __name__ == "__main__":
-    run_multi(
-        [
-            setup_fast_ln,
-            setup_fused_ln,
-            setup_causal_conv1d,
-            setup_selective_scan,
-            setup_paddle_bwd_ops,
-        ],
-    )
+    setup_fast_ln
+    setup_fused_ln
+    setup_causal_conv1d
+    setup_selective_scan
+    setup_paddle_bwd_ops

From 426dd021b49eddc919d639e70f2a2cfc5dc5641b Mon Sep 17 00:00:00 2001
From: YUNSHEN XIE <1084314248@qq.com>
Date: Wed, 3 Sep 2025 22:00:11 +0800
Subject: [PATCH 2/2] Fix function calls in setup.py

---
 ops/csrc/setup.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/ops/csrc/setup.py b/ops/csrc/setup.py
index be0c9584a732..dc6938b7b4fd 100644
--- a/ops/csrc/setup.py
+++ b/ops/csrc/setup.py
@@ -237,13 +237,14 @@ def setup_paddle_bwd_ops():
         ext_modules=CUDAExtension(
             include_dirs=paddle_includes,
             sources=sources,
+            extra_compile_args={}
         ),
     )
 
 
 if __name__ == "__main__":
-    setup_fast_ln
-    setup_fused_ln
-    setup_causal_conv1d
-    setup_selective_scan
-    setup_paddle_bwd_ops
+    setup_fast_ln()
+    setup_fused_ln()
+    setup_causal_conv1d()
+    setup_selective_scan()
+    setup_paddle_bwd_ops()