From 64359b4e35112f3690bf33b766fb654fbf6c3443 Mon Sep 17 00:00:00 2001 From: Marius Brehler Date: Thu, 9 Jan 2025 20:39:28 +0100 Subject: [PATCH] Drop unneeded pip flags and switch torch channel (#792) Drops `--pre` and `--upgrade` where dependencies are pinned to specific versions. Further switches to unstall torch cpu form the stable release channel. --- .github/workflows/ci-sglang-benchmark.yml | 2 +- pytorch-cpu-requirements.txt | 3 +-- requirements-iree-pinned.txt | 1 - 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/.github/workflows/ci-sglang-benchmark.yml b/.github/workflows/ci-sglang-benchmark.yml index c958c1f9c..111535712 100644 --- a/.github/workflows/ci-sglang-benchmark.yml +++ b/.github/workflows/ci-sglang-benchmark.yml @@ -69,7 +69,7 @@ jobs: pip install --no-compile -r requirements.txt -e sharktank/ shortfin/ # Pin to known-working versions. - pip install -f https://iree.dev/pip-release-links.html --pre --upgrade \ + pip install -f https://iree.dev/pip-release-links.html iree-base-compiler==3.1.0rc20241220 \ iree-base-runtime==3.1.0rc20241220 \ "numpy<2.0" diff --git a/pytorch-cpu-requirements.txt b/pytorch-cpu-requirements.txt index aae0297db..4447d2385 100644 --- a/pytorch-cpu-requirements.txt +++ b/pytorch-cpu-requirements.txt @@ -1,3 +1,2 @@ ---pre ---index-url https://download.pytorch.org/whl/test/cpu +--index-url https://download.pytorch.org/whl/cpu/ torch==2.3.0 diff --git a/requirements-iree-pinned.txt b/requirements-iree-pinned.txt index fa7b1fe05..a721d7ebb 100644 --- a/requirements-iree-pinned.txt +++ b/requirements-iree-pinned.txt @@ -1,7 +1,6 @@ # Pinned versions of IREE dependencies. # Keep these versions synced with SHORTFIN_IREE_GIT_TAG in shortfin/CMakeLists.txt ---pre --find-links https://iree.dev/pip-release-links.html iree-base-compiler==3.2.0rc20250109 iree-base-runtime==3.2.0rc20250109