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

README: mention installing nightly pytorch.

pull/209/head
Sean Silva 2021-04-27 11:37:09 -07:00 committed by GitHub
parent 544cb4ef54
commit 390d39a96c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@ -123,9 +123,12 @@ ninja check-npcomp
source .env source .env
``` ```
### PyTorch Frontend (with PyTorch installed via conda) ### PyTorch Frontend
```shell ```shell
# Install PyTorch. We currently track and require the nighly build.
pip3 install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
# Build/test npcomp.
./build_tools/cmake_configure.sh ./build_tools/cmake_configure.sh
cmake --build build --target check-npcomp check-frontends-pytorch cmake --build build --target check-npcomp check-frontends-pytorch
``` ```