upd

Author: yzh119

flashinfer-cubin/build_backend.py

@@ -6,7 +6,6 @@
 import sys
 from pathlib import Path
 from setuptools import build_meta as _orig
-from setuptools.build_meta import *
 
 # Add parent directory to path to import artifacts module
 sys.path.insert(0, str(Path(__file__).parent.parent))
@@ -93,5 +92,17 @@ def build_editable(wheel_directory, config_settings=None, metadata_directory=Non
 # 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
-prepare_metadata_for_build_wheel = _orig.prepare_metadata_for_build_wheel
-prepare_metadata_for_build_editable = _orig.prepare_metadata_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
+    )

flashinfer-jit-cache/build_backend.py

@@ -37,15 +37,15 @@ def _create_build_metadata():
         with open(version_file, "r") as f:
             version = f.read().strip()
     else:
-        version = "0.0.0"
+        version = "0.0.0+unknown"
 
     # Add dev suffix if specified
     dev_suffix = os.environ.get("FLASHINFER_DEV_RELEASE_SUFFIX", "")
     if dev_suffix:
         version = f"{version}.dev{dev_suffix}"
 
     # Get git version
-    git_version = get_git_version()
+    git_version = get_git_version(cwd=Path(__file__).parent.parent)
 
     # Append CUDA version suffix if available
     cuda_suffix = os.environ.get("CUDA_VERSION_SUFFIX", "")
@@ -63,7 +63,7 @@ def _create_build_metadata():
     return version
 
 
-def compile_jit_cache(output_dir: Path, verbose: bool = True):
+def _compile_jit_cache(output_dir: Path, verbose: bool = True):
     """Compile AOT modules using flashinfer.aot functions directly."""
     from flashinfer import aot
 
@@ -91,7 +91,7 @@ def _build_aot_modules():
 
     try:
         # Compile AOT modules
-        compile_jit_cache(aot_package_dir)
+        _compile_jit_cache(aot_package_dir)
 
         # Verify that some modules were actually compiled
         so_files = list(aot_package_dir.rglob("*.so"))
@@ -198,6 +198,16 @@ def prepare_metadata_for_build_wheel(metadata_directory, config_settings=None):
         )
 
 
+def prepare_metadata_for_build_editable(metadata_directory, config_settings=None):
+    """Prepare metadata for editable install, creating build metadata first."""
+    _create_build_metadata()
+
+    with _MonkeyPatchBdistWheel():
+        return _orig.prepare_metadata_for_build_editable(
+            metadata_directory, config_settings
+        )
+
+
 # Export the required interface
 get_requires_for_build_wheel = _orig.get_requires_for_build_wheel
 get_requires_for_build_editable = getattr(