Skip to content

Commit c237795

Browse files
author
chengduo
authored
Reopen test_parallel_executor_transformer (#12452)
* reopen test_parallel_executor_transformer * save the CPU runing time
1 parent b5a3e40 commit c237795

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

python/paddle/fluid/tests/unittests/test_parallel_executor_transformer.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -167,10 +167,9 @@ def setUpClass(cls):
167167
writer.append_tensor(t)
168168
writer.complete_append_tensor()
169169

170-
@unittest.skip("transformer is buggy in multi gpu")
171170
def test_main(self):
172171
self.check_network_convergence(transformer, use_cuda=True)
173-
self.check_network_convergence(transformer, use_cuda=False)
172+
self.check_network_convergence(transformer, use_cuda=False, iter=5)
174173

175174

176175
if __name__ == '__main__':

0 commit comments

Comments
 (0)