Update .travis.yml
This commit is contained in:
parent
fab84910ea
commit
9ee7fe3d3c
18
.travis.yml
18
.travis.yml
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue