upd

flashinfer-ai · Dec 24, 2024 · 48e8d16 · 48e8d16
1 parent 63fa200
commit 48e8d16
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/release_wheel.yml b/.github/workflows/release_wheel.yml
@@ -27,7 +27,7 @@ jobs:
       matrix:
         python: ["3.8", "3.9", "3.10", "3.11", "3.12"]
         cuda: ["11.8", "12.1", "12.4"]
-        torch: ["2.2", "2.3", "2.4"]
+        torch: ["2.3", "2.4", "2.5"]
         exclude: # for cuda 12.4, we only support torch 2.4+
           - cuda: "12.4"
             torch: "2.2"

diff --git a/scripts/update_whl_index.py b/scripts/update_whl_index.py
@@ -6,7 +6,7 @@
     with open(path, "rb") as f:
         sha256 = hashlib.sha256(f.read()).hexdigest()
     ver, cu, torch = re.findall(
-        r"flashinfer-([0-9.]+)\+cu(\d+)torch([0-9.]+)-", path.name
+        r"flashinfer-([0-9.]+(?:\.post[0-9]+)?)\+cu(\d+)torch([0-9.]+)-", path.name
     )[0]
     index_dir = pathlib.Path(f"flashinfer-whl/cu{cu}/torch{torch}/flashinfer")
     index_dir.mkdir(exist_ok=True)