Updated travis and renamed tbptt test gpu -> cuda

pytorch · vfdev-5 · Oct 24, 2019 · Aug 1, 2019 · Aug 1, 2019 · Aug 1, 2019
commit f4a3d4b58bff98f8e0bd14d01c510759279e6bdc
diff --git a/.travis.yml b/.travis.yml
@@ -43,6 +43,9 @@ install:
 
 script:
   - CUDA_VISIBLE_DEVICES="" py.test --tx 4*popen//python=python$TRAVIS_PYTHON_VERSION --cov ignite --cov-report term-missing -vvv tests/
+  # Run test on cuda device
+  # As no GPUs on travis -> all tests will be skipped
+  - CUDA_VISIBLE_DEVICES=0 py.test --cov ignite --cov-append --cov-report term-missing -vvv tests/ -k "on_cuda"
 
   # Smoke tests for the examples
   # Mnist
@@ -75,7 +78,7 @@ script:
   # As no GPUs on travis -> all tests will be skipped
   # 2 is the number of processes <-> number of available GPUs
   - export WORLD_SIZE=2
-  - pytest --dist=each --tx $WORLD_SIZE*popen//python=python$TRAVIS_PYTHON_VERSION tests -m distributed -vvv
+  - py.test --cov ignite --cov-append --cov-report term-missing --dist=each --tx $WORLD_SIZE*popen//python=python$TRAVIS_PYTHON_VERSION tests -m distributed -vvv
 
 after_success:
   - codecov

diff --git a/tests/ignite/contrib/engines/test_tbptt.py b/tests/ignite/contrib/engines/test_tbptt.py
@@ -139,5 +139,5 @@ def test_create_supervised_tbptt_trainer_with_cpu():
 
 
 @pytest.mark.skipif(not torch.cuda.is_available(), reason="Skip if no GPU")
-def test_create_supervised_tbptt_trainer_with_gpu():
+def test_create_supervised_tbptt_trainer_on_cuda():
     _test_create_supervised_tbptt_trainer("cuda")