Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

(Shark 1.0) Fix setup_venv.ps1 not finding pinned torch-mlir #2062

Conversation

one-lithe-rune
Copy link
Contributor

Addresses #2061

Motivation

The version of torch-mlir pinned by d35288e seems to have dropped off of https://llvm.github.io/torch-mlir/package-index/ so .\setup-venv.ps1 fails to install it.

Changes

  • Adds a local file equivalent of https://llvm.github.io/torch-mlir/package-index/ containing links to the pinned torch_mlir versions (links taken from the torch-mlir releases github page).
  • Use the local file instead of the real package index in setup_venv.ps1.

Possible Problems/Concerns

  • I would be entirely unsurprised if this is completely the wrong way to handle this.
  • Does change anything for the setup-venv.sh script as the original commit didn't touch that.

* Adds a local file equivalent of https://llvm.github.io/torch-mlir/package-index/
containing links to the pinned torch_mlir versions (links taken from the torch-mlir
releases github page).
* Use the local file instead of the real package index in setup_venv.ps1.
@monorimet
Copy link
Collaborator

Awesome! Thanks for this.

@monorimet
Copy link
Collaborator

It may need some tweaking for Linux and/or exe builds but generally a good start.

Copy link
Collaborator

@monorimet monorimet left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@monorimet monorimet merged commit bb43181 into nod-ai:SHARK-1.0 Jan 11, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants