experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Reenable MWMS CTL correctness test with multiple GPUs.

Browse Source 
PiperOrigin-RevId: 325245429
Change-Id: Ib4b08c280ac35cdb85b33f6c2779d931eb963e72
This commit is contained in:
Ayush Dubey 2020-08-06 09:24:59 -07:00 committed by TensorFlower Gardener
parent c414595556
commit d029f2f799
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/keras/distribute/ctl_correctness_test.py
View File

@ -252,11 +252,6 @@ class TestDistributionStrategyDnnCorrectness(test.TestCase,
if ('CollectiveAllReduce' in type(distribution).__name__ and
test_util.is_xla_enabled()):
self.skipTest('XLA tests fail with MWMS.')
# Unable to use required_gpus to check if this is a multiGPU combination
# since required_gpus and NamedDistribution cannot be used together.
if ('CollectiveAllReduce' in type(distribution).__name__
and not inside_func and iteration_type == 'dataset'):
self.skipTest('MWMS tests fail with multiple GPUs.')
self.dnn_correctness(distribution, optimizer_fn, iteration_type,
inside_func, sync_batchnorm)