File tree Expand file tree Collapse file tree 1 file changed +6
-23
lines changed
Expand file tree Collapse file tree 1 file changed +6
-23
lines changed Original file line number Diff line number Diff line change @@ -53,20 +53,6 @@ def run_single(func):
5353 p .join ()
5454
5555
56- def run_multi (func_list ):
57- processes = []
58- for func in func_list :
59- processes .append (multiprocessing .Process (target = func ))
60- processes .append (multiprocessing .Process (target = func ))
61- processes .append (multiprocessing .Process (target = func ))
62-
63- for p in processes :
64- p .start ()
65-
66- for p in processes :
67- p .join ()
68-
69-
7056cc_flag = get_gencode_flags (compiled_all = False )
7157cc = get_sm_version ()
7258
@@ -251,17 +237,14 @@ def setup_paddle_bwd_ops():
251237 ext_modules = CUDAExtension (
252238 include_dirs = paddle_includes ,
253239 sources = sources ,
240+ extra_compile_args = {}
254241 ),
255242 )
256243
257244
258245if __name__ == "__main__" :
259- run_multi (
260- [
261- setup_fast_ln ,
262- setup_fused_ln ,
263- setup_causal_conv1d ,
264- setup_selective_scan ,
265- setup_paddle_bwd_ops ,
266- ],
267- )
246+ setup_fast_ln ()
247+ setup_fused_ln ()
248+ setup_causal_conv1d ()
249+ setup_selective_scan ()
250+ setup_paddle_bwd_ops ()
You can’t perform that action at this time.
0 commit comments