Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
[MXAPPS-581] Move notebooks requiring multi-gpus.
Browse files Browse the repository at this point in the history
Move two notebooks requiring multi-GPUs out of the single GPU test suite.
  • Loading branch information
vishaalkapoor committed Jul 24, 2018
1 parent cce0ca7 commit 47d0e08
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 10 deletions.
4 changes: 4 additions & 0 deletions tests/nightly/straight_dope/test_notebooks_multi_gpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,10 @@ def setUpClass(self):
def test_multiple_gpus_gluon(self):
assert _test_notebook('chapter07_distributed-learning/multiple-gpus-gluon')

# TODO(vishaalk): Do a dry run, and then enable.
#def test_training_with_multiple_machines(self):
# assert _test_notebook('chapter07_distributed-learning/training-with-multiple-machines')

# Chapter 8

# TODO(vishaalk): Module skimage needs to be added to docker image.
Expand Down
13 changes: 3 additions & 10 deletions tests/nightly/straight_dope/test_notebooks_single_gpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,10 +32,12 @@
'chapter01_crashcourse/introduction',
'chapter01_crashcourse/chapter-one-problem-set',
'chapter02_supervised-learning/environment',
'chapter07_distributed-learning/multiple-gpus-scratch',
'chapter07_distributed-learning/multiple-gpus-gluon',
'chapter07_distributed-learning/training-with-multiple-machines'
]



class StraightDopeSingleGpuTests(unittest.TestCase):
@classmethod
def setUpClass(self):
Expand Down Expand Up @@ -226,15 +228,6 @@ def test_adam_gluon(self):
def test_hybridize(self):
assert _test_notebook('chapter07_distributed-learning/hybridize')

# TODO(vishaalk): module 'mxnet.gluon' has no attribute 'autograd'
#def test_multiple_gpus_scratch(self):
# assert _test_notebook('chapter07_distributed-learning/multiple-gpus-scratch')

def test_multiple_gpus_gluon(self):
assert _test_notebook('chapter07_distributed-learning/multiple-gpus-gluon')

def test_training_with_multiple_machines(self):
assert _test_notebook('chapter07_distributed-learning/training-with-multiple-machines')

# Chapter 8

Expand Down

0 comments on commit 47d0e08

Please sign in to comment.