llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix feedback from Anush.

pull/355/head
Sean Silva 2021-10-07 01:28:13 +00:00
parent a157d6c946
commit de8ea59d52
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
README.md
View File

@ -37,17 +37,18 @@ We have few paths to lower down to the Torch MLIR Dialect.
git clone https://github.com/llvm/torch-mlir
cd torch-mlir
python -m venv mlir_venv
source ./mlir_venv/bin/activate
# Some older pip installs may not be able to handle the recent PyTorch deps
./mlir_venv/bin/python -m pip install --upgrade pip
python -m pip install --upgrade pip
# Install Torch deps.
./mlir_venv/bin/python -m pip install --pre torch torchvision pybind11 -f "https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html"
python -m pip install --pre torch torchvision pybind11 -f "https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html"
# Install torch-mlir packages.
./mlir_venv/bin/python -m pip install torch_mlir -f "https://github.com/llvm/torch-mlir/releases"
python -m pip install torch_mlir -f "https://github.com/llvm/torch-mlir/releases"
# Install some deps for the example
./mlir_venv/bin/python -m pip install requests pillow
python -m pip install requests pillow
# Run it!
./mlir_venv/bin/python examples/torchscript_resnet18_e2e.py
python examples/torchscript_resnet18_e2e.py
```
## Check out the code