upd

Author: yzh119

build_backend.py

@@ -25,20 +25,48 @@
 _data_dir = _root / "flashinfer" / "data"
 
 
-def get_version():
-    package_version = (_root / "version.txt").read_text().strip()
+def _create_build_metadata():
+    """Create build metadata file with version information."""
+    version_file = _root / "version.txt"
+    if version_file.exists():
+        with open(version_file, "r") as f:
+            version = f.read().strip()
+    else:
+        version = "0.0.0+unknown"
+
+    # Add dev suffix if specified
     dev_suffix = os.environ.get("FLASHINFER_DEV_RELEASE_SUFFIX", "")
     if dev_suffix:
-        package_version = f"{package_version}.dev{dev_suffix}"
-    return package_version
+        version = f"{version}.dev{dev_suffix}"
 
+    # Get git version
+    git_version = get_git_version(cwd=_root)
 
-# Create _build_meta.py at import time so setuptools can read the version
-build_meta_file = _root / "flashinfer" / "_build_meta.py"
-with open(build_meta_file, "w") as f:
-    f.write('"""Build metadata for flashinfer package."""\n')
-    f.write(f'__version__ = "{get_version()}"\n')
-    f.write(f'__git_version__ = "{get_git_version(cwd=_root)}"\n')
+    # Create build metadata in the source tree
+    package_dir = Path(__file__).parent / "flashinfer"
+    build_meta_file = package_dir / "_build_meta.py"
+
+    # Check if we're in a git repository
+    git_dir = Path(__file__).parent / ".git"
+    in_git_repo = git_dir.exists()
+
+    # If file exists and not in git repo (installing from sdist), keep existing file
+    if build_meta_file.exists() and not in_git_repo:
+        print("Build metadata file already exists (not in git repo), keeping it")
+        return version
+
+    # In git repo (editable) or file doesn't exist, create/update it
+    with open(build_meta_file, "w") as f:
+        f.write('"""Build metadata for flashinfer package."""\n')
+        f.write(f'__version__ = "{version}"\n')
+        f.write(f'__git_version__ = "{git_version}"\n')
+
+    print(f"Created build metadata file with version {version}")
+    return version
+
+
+# Create build metadata as soon as this module is imported
+_create_build_metadata()
 
 
 def write_if_different(path: Path, content: str) -> None:
@@ -74,11 +102,6 @@ def ln(source: str, target: str) -> None:
     ln("csrc", "csrc")
     ln("include", "include")
 
-    # Always ensure version.txt is present
-    version_file = _data_dir / "version.txt"
-    if not version_file.exists() or not use_symlinks:
-        shutil.copy(_root / "version.txt", version_file)
-
 
 def _prepare_for_wheel():
     # For wheel, copy actual files instead of symlinks so they are included in the wheel

docs/conf.py

@@ -1,4 +1,3 @@
-from pathlib import Path
 from typing import Any, List
 
 import flashinfer  # noqa: F401
@@ -12,7 +11,6 @@
 # -- Project information -----------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
 
-root = Path(__file__).parents[1].resolve()
 # FlashInfer is installed via pip before building docs
 autodoc_mock_imports = [
     "torch",
@@ -28,9 +26,8 @@
 author = "FlashInfer Contributors"
 copyright = f"2023-2025, {author}"
 
-package_version = (root / "version.txt").read_text().strip()
-version = package_version
-release = package_version
+version = flashinfer.__version__
+release = flashinfer.__version__
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

flashinfer-cubin/build_backend.py

@@ -66,6 +66,16 @@ def _create_build_metadata():
     package_dir = Path(__file__).parent / "flashinfer_cubin"
     build_meta_file = package_dir / "_build_meta.py"
 
+    # Check if we're in a git repository
+    git_dir = Path(__file__).parent.parent / ".git"
+    in_git_repo = git_dir.exists()
+
+    # If file exists and not in git repo (installing from sdist), keep existing file
+    if build_meta_file.exists() and not in_git_repo:
+        print("Build metadata file already exists (not in git repo), keeping it")
+        return version
+
+    # In git repo (editable) or file doesn't exist, create/update it
     with open(build_meta_file, "w") as f:
         f.write('"""Build metadata for flashinfer-cubin package."""\n')
         f.write(f'__version__ = "{version}"\n')

flashinfer-jit-cache/build_backend.py

@@ -54,6 +54,16 @@ def _create_build_metadata():
         version = f"{version}+{cuda_suffix}"
     build_meta_file = Path(__file__).parent / "flashinfer_jit_cache" / "_build_meta.py"
 
+    # Check if we're in a git repository
+    git_dir = Path(__file__).parent.parent / ".git"
+    in_git_repo = git_dir.exists()
+
+    # If file exists and not in git repo (installing from sdist), keep existing file
+    if build_meta_file.exists() and not in_git_repo:
+        print("Build metadata file already exists (not in git repo), keeping it")
+        return version
+
+    # In git repo (editable) or file doesn't exist, create/update it
     with open(build_meta_file, "w") as f:
         f.write('"""Build metadata for flashinfer-jit-cache package."""\n')
         f.write(f'__version__ = "{version}"\n')

pyproject.toml

@@ -61,8 +61,7 @@ exclude = ["flashinfer-jit-cache*", "flashinfer-cubin*"]
 ]
 "flashinfer.data" = [
     "csrc/**",
-    "include/**",
-    "version.txt"
+    "include/**"
 ]
 "flashinfer.data.cutlass" = [
     "include/**",