Kimirock/PaddleNLP
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix custom op compile bug (#11064)

Browse Source 
* fix bug

* Fix function calls in setup.py

---------

Co-authored-by: liym27 <33742067+liym27@users.noreply.github.com>
Co-authored-by: YUNSHEN XIE <1084314248@qq.com>
This commit is contained in:
risemeup1
2025-09-04 11:05:56 +08:00
committed by GitHub
parent 1060ed47e7
commit 96a39777e3
1 changed files with 6 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
ops/csrc/setup.py
View File

@@ -53,20 +53,6 @@ def run_single(func):
p.join() 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_flag = get_gencode_flags(compiled_all=False)
cc = get_sm_version() cc = get_sm_version()
@@ -251,17 +237,14 @@ def setup_paddle_bwd_ops():
ext_modules=CUDAExtension( ext_modules=CUDAExtension(
include_dirs=paddle_includes, include_dirs=paddle_includes,
sources=sources, sources=sources,
extra_compile_args={}
), ),
) )
if __name__ == "__main__": if __name__ == "__main__":
run_multi( setup_fast_ln()
[ setup_fused_ln()
setup_fast_ln, setup_causal_conv1d()
setup_fused_ln, setup_selective_scan()
setup_causal_conv1d, setup_paddle_bwd_ops()
setup_selective_scan,
setup_paddle_bwd_ops,
],
)