diff --git a/.github/workflows/release_wheel.yml b/.github/workflows/release_wheel.yml index 9fbbf8bf..3e33b10d 100644 --- 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 index 21f2ec97..57f5f939 100644 --- 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)