Update .travis.yml

This commit is contained in:
Martin Kroeker 2022-09-09 13:23:27 +02:00 committed by GitHub
parent fab84910ea
commit 9ee7fe3d3c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 4 deletions

View File

@ -13,6 +13,8 @@ matrix:
apt:
packages:
- gfortran
- g++
- conda
# before_script: &common-before
# - COMMON_FLAGS="DYNAMIC_ARCH=1 TARGET=NEHALEM NUM_THREADS=32"
# script:
@ -28,12 +30,20 @@ matrix:
os: linux
arch: ppc64le
before_script: &common-before
- COMMON_FLAGS="DYNAMIC_ARCH=1 TARGET=POWER8 NUM_THREADS=32"
- COMMON_FLAGS="DYNAMIC_ARCH=1 TARGET=POWER8"
script:
- travis_wait 40 make QUIET_MAKE=1 $COMMON_FLAGS $BTYPE
- make -C test $COMMON_FLAGS $BTYPE
- make -C ctest $COMMON_FLAGS $BTYPE
- make -C utest $COMMON_FLAGS $BTYPE
- conda install astunparse numpy ninja pyyaml setuptools cmake cffi typing_extensions future six requests dataclasses
- conda init bash
- conda create condas
- conda activate condas
- git clone --recursive https://github.com/pytorch/pytorch
- export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
- cd pytorch
- python setup.py install
- cd test
- sed -i -e "s/@onlyCUDA//g" -e "s/@largeTensorTest('12GB')//g" test_nn.py
- python test_nn.py TestNNDeviceTypeCPU.test_conv_large_cpu
env:
# for matrix annotation only
- TARGET_BOX=PPC64LE_LINUX