nccl pass parallel_do test

yu239-patch-1
Yang Yang 7 years ago
parent 0d57ca46ea
commit bb3ae20664

@ -172,12 +172,18 @@ class ParallelOpTest(BaseParallelForTest):
loss = fluid.layers.mean(x=hidden)
yield loss
def test_fc_with_tiny_data(self):
def test_simple_fc(self):
self.run_test(
callback=self.__network__,
feed={'img': numpy.random.random(size=(8, 784)).astype('float32')},
fetch=['fc1.w@GRAD'])
def test_fc_with_tiny_data(self):
self.run_test(
callback=self.__network__,
feed={'img': numpy.random.random(size=(1, 784)).astype('float32')},
fetch=['fc1.w@GRAD'])
class ParallelOpTestMultipleInput(BaseParallelForTest):
@staticmethod

Loading…
Cancel
Save