Skip to content

Commit 1b676be

Browse files
authored
Merge pull request #13361 from JiabinYang/mac/unittest428
fix test_parallel_executor_transformer
2 parents 82236a0 + bdd957b commit 1b676be

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
from parallel_executor_test_base import TestParallelExecutorBase
2121
import unittest
2222
import paddle
23+
import paddle.fluid.core as core
2324
import paddle.dataset.wmt16 as wmt16
2425
import os
2526

@@ -170,7 +171,8 @@ def setUpClass(cls):
170171
writer.complete_append_tensor()
171172

172173
def test_main(self):
173-
self.check_network_convergence(transformer, use_cuda=True)
174+
if core.is_compiled_with_cuda():
175+
self.check_network_convergence(transformer, use_cuda=True)
174176
self.check_network_convergence(transformer, use_cuda=False, iter=5)
175177

176178

0 commit comments

Comments
 (0)