Update .travis.yml

This commit is contained in:
Martin Kroeker 2022-09-10 10:49:04 +02:00 committed by GitHub
parent 34ae1a0aca
commit 03c9ffd81f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions

View File

@ -39,14 +39,17 @@ matrix:
- conda init
- #conda create condas
- conda activate #condas
- conda install -y astunparse numpy ninja pyyaml setuptools cmake cffi typing_extensions future six requests dataclasses
- conda install -y astunparse numpy ninja pyyaml setuptools cmake cffi typing_extensions future six requests dataclasses torch hypothesis
- git clone --recursive https://github.com/pytorch/pytorch
- export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
- #export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
- cd pytorch
- python3 setup.py install
- #python3 setup.py install
- cd test
- sed -i -e "s/@onlyCUDA//g" -e "s/@largeTensorTest('12GB')//g" test_nn.py
- python3 test_nn.py TestNNDeviceTypeCPU.test_conv_large_cpu
- cp ../../libopenblas.so $HOME/conda/lib
- cp ../../libopenblas.so.0 $HOME/conda/lib
- python3 test_nn.py TestNNDeviceTypeCPU.test_conv_large_cpu
env:
# for matrix annotation only
- TARGET_BOX=PPC64LE_LINUX