fix windows python issue (freeze_support)

thehajime · Dec 26, 2024 · b20316d · b20316d
1 parent faf517b
commit b20316d
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/arch/lkl/scripts/headers_install.py b/arch/lkl/scripts/headers_install.py
@@ -190,11 +190,13 @@ def process_header(h):
     print("  REPLACE\t%s" % (out_dir + "/" + os.path.basename(h)))
     replace(h)
 
-p = multiprocessing.Pool(args.jobs)
-try:
-    p.map_async(process_header, headers).wait(999999)
-    p.close()
-except:
-    p.terminate()
-finally:
-    p.join()
+if __name__ == '__main__':
+    multiprocessing.freeze_support()
+    p = multiprocessing.Pool(args.jobs)
+    try:
+        p.map_async(process_header, headers).wait(999999)
+        p.close()
+    except:
+        p.terminate()
+    finally:
+        p.join()