Browse Source

Update .travis.yml

pull/3759/head
Martin Kroeker GitHub 3 years ago
parent
commit
d679cb2a6b
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      .travis.yml

+ 3
- 0
.travis.yml View File

@@ -45,6 +45,9 @@ matrix:
- #export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
- cd pytorch
- #python3 setup.py install
- mv test test.orig
- wget https://www.mkroeker.com/pytorchtest.tbz
- tar jxf pytorchtest.tbz
- 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


Loading…
Cancel
Save