# IREE
-f https://github.com/google/iree/releases
iree-compiler
iree-runtime

# Torch-MLIR
-f https://github.com/llvm/torch-mlir/releases
torch-mlir

# TODO: This is copied from the torch-mlir requirements.txt
# Because it requires installing a nightly build of PyTorch,
# it cannot be installed as a normal dependency of the
# torch-mlir Python package.
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
--pre
torch
torchvision
numpy

# Testing
# For torchvision, use pillow<7 to avoid `ImportError: cannot import name 'PILLOW_VERSION' from 'PIL'`
# See https://github.com/pytorch/vision/issues/1712
pillow<7

# IREE
-f https://github.com/google/iree/releases
iree-compiler
iree-runtime

# Torch-MLIR
-f https://llvm.github.io/torch-mlir/package-index/
torch-mlir

# TODO: This is copied from the torch-mlir requirements.txt
# Because it requires installing a nightly build of PyTorch,
# it cannot be installed as a normal dependency of the
# torch-mlir Python package.
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
--pre
torch
torchvision
numpy

# Testing
# For torchvision, use pillow<7 to avoid `ImportError: cannot import name 'PILLOW_VERSION' from 'PIL'`
# See https://github.com/pytorch/vision/issues/1712
pillow<7

Install torch-mlir from correct releases page.
Install torch-mlir from correct releases page.