upd

Author: yzh119

flashinfer-cubin/build_backend.py

@@ -13,18 +13,8 @@
 
 from build_utils import get_git_version
 
-
-# add flashinfer._build_meta, always override to ensure version is up-to-date
-build_meta_file = Path(__file__).parent.parent / "flashinfer" / "_build_meta.py"
-version_file = Path(__file__).parent.parent / "version.txt"
-if version_file.exists():
-    with open(version_file, "r") as f:
-        version = f.read().strip()
-git_version = get_git_version(cwd=Path(__file__).parent.parent)
-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')
+# Skip version check when building flashinfer-cubin package
+os.environ["FLASHINFER_DISABLE_VERSION_CHECK"] = "1"
 
 
 def _download_cubins():

flashinfer-jit-cache/build_backend.py

@@ -26,18 +26,8 @@
 
 from build_utils import get_git_version
 
-
-# add flashinfer._build_meta, always override to ensure version is up-to-date
-build_meta_file = Path(__file__).parent.parent / "flashinfer" / "_build_meta.py"
-version_file = Path(__file__).parent.parent / "version.txt"
-if version_file.exists():
-    with open(version_file, "r") as f:
-        version = f.read().strip()
-git_version = get_git_version(cwd=Path(__file__).parent.parent)
-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')
+# Skip version check when building flashinfer-jit-cache package
+os.environ["FLASHINFER_DISABLE_VERSION_CHECK"] = "1"
 
 
 def _create_build_metadata():

flashinfer/jit/env.py

@@ -44,11 +44,16 @@ def _get_cubin_dir():
         import flashinfer_cubin
 
         flashinfer_cubin_version = flashinfer_cubin.__version__
-        if flashinfer_version != flashinfer_cubin_version:
+        # Allow bypassing version check with environment variable
+        if (
+            not os.getenv("FLASHINFER_DISABLE_VERSION_CHECK")
+            and flashinfer_version != flashinfer_cubin_version
+        ):
             raise RuntimeError(
                 f"flashinfer-cubin version ({flashinfer_cubin_version}) does not match "
                 f"flashinfer version ({flashinfer_version}). "
-                "Please install the same version of both packages."
+                "Please install the same version of both packages. "
+                "Set FLASHINFER_DISABLE_VERSION_CHECK=1 to bypass this check."
             )
 
         return pathlib.Path(flashinfer_cubin.get_cubin_dir())
@@ -78,11 +83,15 @@ def _get_aot_dir():
         flashinfer_jit_cache_version = flashinfer_jit_cache.__version__
         # NOTE(Zihao): we don't use exact version match here because the version of flashinfer-jit-cache
         # contains the CUDA version suffix: e.g. 0.3.1+cu129.
-        if not flashinfer_jit_cache_version.startswith(flashinfer_version):
+        # Allow bypassing version check with environment variable
+        if not os.getenv(
+            "FLASHINFER_DISABLE_VERSION_CHECK"
+        ) and not flashinfer_jit_cache_version.startswith(flashinfer_version):
             raise RuntimeError(
                 f"flashinfer-jit-cache version ({flashinfer_jit_cache_version}) does not match "
                 f"flashinfer version ({flashinfer_version}). "
-                "Please install the same version of both packages."
+                "Please install the same version of both packages. "
+                "Set FLASHINFER_DISABLE_VERSION_CHECK=1 to bypass this check."
             )
 
         return pathlib.Path(flashinfer_jit_cache.get_jit_cache_dir())