wheels: trim setup.py comments and move LICENSE copy near setup()

The post-install cleanup blocks were carrying redundant inline
annotations that just listed what the dev_paths literal already names.
Collapse the static-archives loop into the same dev_paths list (one
audience filter, one pass) and shrink the comment to the actual
"why".

Also relocate the LICENSE materialization from the module top to
immediately before setup(), so the side-effect-at-import sits next to
the license_files= consumer that needs it. Cannot move into
CMakeBuild.run() — egg_info reads license_files before build_ext fires.

Author: hunhoffe

utils/mlir_aie_wheels/setup.py

@@ -305,47 +305,33 @@ def build_extension(self, ext: CMakeExtension) -> None:
                 check=True,
             )
 
-            # Static archives (lib/*.a) only matter to anyone linking C++ code
-            # against AIE/MLIR. PyPI consumers use the Python API, never these.
-            for archive in (Path(install_dir) / "lib").glob("*.a"):
-                archive.unlink()
-
-            # Same audience filter for the dev-only headers + cmake configs:
-            #   include/aie/                  - MLIR C++ pass-author headers
-            #   include/aie-c/                - C API headers
-            #   include/bootgen_c_api.h       - bootgen C API header
-            #   include/xaienginecdo_static/  - xaiengine headers, paired with
-            #                                   the static archives we just dropped
-            #   lib/cmake/                    - find_package(AIE) configs for
-            #                                   downstream native builds
-            # User-kernel-facing headers (aie_api/, aie_kernels/) stay — user
-            # AIE kernels #include those at compile time.
+            # C++-developer-only content: anyone pip installing this wheel
+            # reaches the toolchain through Python, never through native
+            # linking. aie_api/ and aie_kernels/ headers stay — user AIE
+            # kernels #include those at compile time.
             dev_paths = [
                 Path(install_dir) / "include" / "aie",
                 Path(install_dir) / "include" / "aie-c",
                 Path(install_dir) / "include" / "bootgen_c_api.h",
                 Path(install_dir) / "include" / "xaienginecdo_static",
                 Path(install_dir) / "lib" / "cmake",
+                *(Path(install_dir) / "lib").glob("*.a"),
             ]
             for p in dev_paths:
                 if p.is_dir():
                     shutil.rmtree(p)
                 elif p.exists():
                     p.unlink()
 
-            # CMake leaked build artifacts into the install prefix:
-            #   src/                  - declare_mlir_python_sources staging,
-            #                           full duplicate of python/
-            #   lib/objects-Release/  - intermediate .o files
-            #   **/__pycache__/       - bytecode caches generated by any python
-            #                           run that happened during build
-            # None of these belong in a shipped wheel.
-            src_staging = Path(install_dir) / "src"
-            if src_staging.exists():
-                shutil.rmtree(src_staging)
-            objects_dir = Path(install_dir) / "lib" / "objects-Release"
-            if objects_dir.exists():
-                shutil.rmtree(objects_dir)
+            # CMake leaks staging directories, intermediate .o files, and
+            # __pycache__ caches into the install prefix; none belong in a
+            # shipped wheel.
+            for leaked in [
+                Path(install_dir) / "src",
+                Path(install_dir) / "lib" / "objects-Release",
+            ]:
+                if leaked.exists():
+                    shutil.rmtree(leaked)
             for pycache in Path(install_dir).rglob("__pycache__"):
                 shutil.rmtree(pycache, ignore_errors=True)
 
@@ -416,10 +402,6 @@ def run(self):
     )
 ).absolute()
 
-_license_src = MLIR_AIE_SOURCE_DIR / "LICENSE"
-if _license_src.exists():
-    shutil.copy(_license_src, Path(__file__).parent / "LICENSE")
-
 
 def parse_requirements(filename):
     with open(filename) as f:
@@ -441,6 +423,10 @@ def parse_requirements(filename):
         return requirements
 
 
+_license = MLIR_AIE_SOURCE_DIR / "LICENSE"
+if _license.exists():
+    shutil.copy(_license, Path(__file__).parent / "LICENSE")
+
 setup(
     name="mlir-aie" if check_env("ENABLE_RTTI", 1) else "mlir-aie-no-rtti",
     version=get_version(),