[Android] Add -j option to cmake build (#2321)

This PR adds the `-j` option to cmake build to parallelize the build job over CPU cores.
mlc-ai · May 10, 2024 · 2b8aadf · 2b8aadf
1 parent 3a0b42c
commit 2b8aadf
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/android/mlc4j/prepare_libs.py b/android/mlc4j/prepare_libs.py
@@ -43,13 +43,31 @@ def run_cmake(mlc4j_path: Path):
 
 def run_cmake_build():
     logger.info("Running cmake build")
-    cmd = ["cmake", "--build", ".", "--target", "tvm4j_runtime_packed", "--config", "release"]
+    cmd = [
+        "cmake",
+        "--build",
+        ".",
+        "--target",
+        "tvm4j_runtime_packed",
+        "--config",
+        "release",
+        f"-j{os.cpu_count()}",
+    ]
     subprocess.run(cmd, check=True, env=os.environ)
 
 
 def run_cmake_install():
     logger.info("Running cmake install")
-    cmd = ["cmake", "--build", ".", "--target", "install", "--config", "release", "-j"]
+    cmd = [
+        "cmake",
+        "--build",
+        ".",
+        "--target",
+        "install",
+        "--config",
+        "release",
+        f"-j{os.cpu_count()}",
+    ]
     subprocess.run(cmd, check=True, env=os.environ)