Update .travis.yml

.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
+        - conda install  astunparse numpy ninja pyyaml setuptools cmake cffi typing_extensions future six requests dataclasses
-        - make -C ctest $COMMON_FLAGS $BTYPE
+        - conda init bash
-        - make -C utest $COMMON_FLAGS $BTYPE
+        - 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