From 398cd2b940a1460efde471a7428d2d5faae07508 Mon Sep 17 00:00:00 2001 From: Zihao Ye Date: Mon, 23 Dec 2024 16:37:10 -0800 Subject: [PATCH] ci: fix the update_whl_index script to regonize version number with "post" and add torch2.5 (#694) Ref #692 --- .github/workflows/release_wheel.yml | 2 +- scripts/update_whl_index.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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)