Use single setup.py and move writing to common module

Author: danieldk

build2cmake/src/templates/cpu/setup.py

@@ -69,13 +69,17 @@ def build_extension(self, ext: CMakeExtension) -> None:
                 "-DCMAKE_CXX_COMPILER_LAUNCHER=sccache",
                 "-DCMAKE_CUDA_COMPILER_LAUNCHER=sccache",
                 "-DCMAKE_HIP_COMPILER_LAUNCHER=sccache",
+                "-DCMAKE_OBJC_COMPILER_LAUNCHER=sccache",
+                "-DCMAKE_OBJCXX_COMPILER_LAUNCHER=sccache",
             ]
         elif is_ccache_available():
             cmake_args += [
                 "-DCMAKE_C_COMPILER_LAUNCHER=ccache",
                 "-DCMAKE_CXX_COMPILER_LAUNCHER=ccache",
                 "-DCMAKE_CUDA_COMPILER_LAUNCHER=ccache",
                 "-DCMAKE_HIP_COMPILER_LAUNCHER=ccache",
+                "-DCMAKE_OBJC_COMPILER_LAUNCHER=ccache",
+                "-DCMAKE_OBJCXX_COMPILER_LAUNCHER=ccache",
             ]
 
         num_jobs = os.getenv("MAX_JOBS", None)
@@ -96,17 +100,13 @@ def build_extension(self, ext: CMakeExtension) -> None:
             logger.info(
                 "Using NVCC_THREADS=%d as the number of nvcc threads.", nvcc_threads
             )
-        else:
-            nvcc_threads = 1
-        num_jobs = max(1, num_jobs // nvcc_threads)
+            num_jobs = max(1, num_jobs // nvcc_threads)
+            cmake_args += ["-DNVCC_THREADS={}".format(nvcc_threads)]
 
         build_args += [f"-j{num_jobs}"]
         if sys.platform == "win32":
             build_args += ["--config", cfg]
 
-        if nvcc_threads:
-            cmake_args += ["-DNVCC_THREADS={}".format(nvcc_threads)]
-
         build_temp = Path(self.build_temp) / ext.name
         if not build_temp.exists():
             build_temp.mkdir(parents=True)
@@ -117,13 +117,13 @@ def build_extension(self, ext: CMakeExtension) -> None:
         subprocess.run(
             ["cmake", "--build", ".", *build_args], cwd=build_temp, check=True
         )
+
         if sys.platform == "win32":
             # Move the dylib one folder up for discovery.
             for filename in os.listdir(extdir / cfg):
                 move(extdir / cfg / filename, extdir / filename)
 
 
-
 {% set python_name = name | replace('-', '_') %}
 setup(
     name="{{ python_name }}",