From b444528715f473f57604f457fd35f742a659c9f4 Mon Sep 17 00:00:00 2001 From: powderluv Date: Fri, 23 Jun 2023 19:19:28 -0700 Subject: [PATCH] Pin torch-mlir for windows too --- setup_venv.ps1 | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup_venv.ps1 b/setup_venv.ps1 index 7427717419..a9d5be98cb 100644 --- a/setup_venv.ps1 +++ b/setup_venv.ps1 @@ -89,7 +89,7 @@ else {python -m venv .\shark.venv\} python -m pip install --upgrade pip pip install wheel pip install -r requirements.txt -pip install --pre torch-mlir torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://llvm.github.io/torch-mlir/package-index/ +pip install --pre torch-mlir==20230620.875 torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://llvm.github.io/torch-mlir/package-index/ pip install --upgrade -f https://nod-ai.github.io/SHARK-Runtime/pip-release-links.html iree-compiler iree-runtime Write-Host "Building SHARK..." pip install -e . -f https://llvm.github.io/torch-mlir/package-index/ -f https://nod-ai.github.io/SHARK-Runtime/pip-release-links.html