From 52a6d29d7736343f5def72dc47ff44f53a3fb3fd Mon Sep 17 00:00:00 2001
From: Houchen Li <houchen_li@hotmail.com>
Date: Mon, 26 May 2025 13:12:00 +0800
Subject: [PATCH] involve parallel compiling

---
 setup_env.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup_env.py b/setup_env.py
index f15d65f4..4c487d9a 100644
--- a/setup_env.py
+++ b/setup_env.py
@@ -1,4 +1,5 @@
 import subprocess
+import multiprocessing
 import signal
 import sys
 import os
@@ -213,7 +214,7 @@ def compile():
     logging.info("Compiling the code using CMake.")
     run_command(["cmake", "-B", "build", *COMPILER_EXTRA_ARGS[arch], *OS_EXTRA_ARGS.get(platform.system(), []), "-DCMAKE_C_COMPILER=clang", "-DCMAKE_CXX_COMPILER=clang++"], log_step="generate_build_files")
     # run_command(["cmake", "--build", "build", "--target", "llama-cli", "--config", "Release"])
-    run_command(["cmake", "--build", "build", "--config", "Release"], log_step="compile")
+    run_command(["cmake", "--build", "build", "--config", "Release", "-j", str(multiprocessing.cpu_count())], log_step="compile")
 
 def main():
     setup_gguf()