upd

Author: yzh119

flashinfer-cubin/build_backend.py

@@ -75,34 +75,24 @@ def _create_build_metadata():
     return version
 
 
+# Create build metadata as soon as this module is imported
+_create_build_metadata()
+
+
 def build_wheel(wheel_directory, config_settings=None, metadata_directory=None):
     """Build a wheel, downloading cubins first."""
     _download_cubins()
-    _create_build_metadata()
     return _orig.build_wheel(wheel_directory, config_settings, metadata_directory)
 
 
 def build_editable(wheel_directory, config_settings=None, metadata_directory=None):
     """Build an editable install, downloading cubins first."""
     _download_cubins()
-    _create_build_metadata()
     return _orig.build_editable(wheel_directory, config_settings, metadata_directory)
 
 
 # Pass through all other hooks
 get_requires_for_build_wheel = _orig.get_requires_for_build_wheel
 get_requires_for_build_editable = _orig.get_requires_for_build_editable
-
-
-def prepare_metadata_for_build_wheel(metadata_directory, config_settings=None):
-    """Prepare metadata for wheel build, creating build metadata first."""
-    _create_build_metadata()
-    return _orig.prepare_metadata_for_build_wheel(metadata_directory, config_settings)
-
-
-def prepare_metadata_for_build_editable(metadata_directory, config_settings=None):
-    """Prepare metadata for editable install, creating build metadata first."""
-    _create_build_metadata()
-    return _orig.prepare_metadata_for_build_editable(
-        metadata_directory, config_settings
-    )
+prepare_metadata_for_build_wheel = _orig.prepare_metadata_for_build_wheel
+prepare_metadata_for_build_editable = _orig.prepare_metadata_for_build_editable

flashinfer-jit-cache/build_backend.py

@@ -63,6 +63,10 @@ def _create_build_metadata():
     return version
 
 
+# Create build metadata as soon as this module is imported
+_create_build_metadata()
+
+
 def _compile_jit_cache(output_dir: Path, verbose: bool = True):
     """Compile AOT modules using flashinfer.aot functions directly."""
     from flashinfer import aot
@@ -107,7 +111,6 @@ def _build_aot_modules():
 
 def _prepare_build():
     """Shared preparation logic for both wheel and editable builds."""
-    _create_build_metadata()
     _build_aot_modules()
 
 
@@ -190,18 +193,14 @@ def build_editable(wheel_directory, config_settings=None, metadata_directory=Non
 
 def prepare_metadata_for_build_wheel(metadata_directory, config_settings=None):
     """Prepare metadata with platform-specific wheel tags."""
-    _create_build_metadata()
-
     with _MonkeyPatchBdistWheel():
         return _orig.prepare_metadata_for_build_wheel(
             metadata_directory, config_settings
         )
 
 
 def prepare_metadata_for_build_editable(metadata_directory, config_settings=None):
-    """Prepare metadata for editable install, creating build metadata first."""
-    _create_build_metadata()
-
+    """Prepare metadata for editable install."""
     with _MonkeyPatchBdistWheel():
         return _orig.prepare_metadata_for_build_editable(
             metadata_directory, config_settings