Update .travis.yml

This commit is contained in:
Martin Kroeker 2022-09-10 15:34:43 +02:00 committed by GitHub
parent 3beff1763c
commit d679cb2a6b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

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