upd

Author: yzh119

.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"

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)