Skip to content

Commit 1001f8e

Browse files
author
chengduo
authored
Add is_compiled_with_cuda for parallel_exe_crf (#14304)
test=develop
1 parent 2f0a379 commit 1001f8e

File tree

1 file changed

+13
-8
lines changed

1 file changed

+13
-8
lines changed

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

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
import paddle.dataset.conll05 as conll05
1818
import paddle.fluid as fluid
19+
import paddle.fluid.core as core
1920
import unittest
2021
import paddle
2122
import numpy as np
@@ -177,32 +178,36 @@ def check_network_convergence(self,
177178
def test_update_sparse_parameter_all_reduce(self):
178179
build_strategy = fluid.BuildStrategy()
179180
build_strategy.reduce_strategy = fluid.BuildStrategy.ReduceStrategy.AllReduce
180-
self.check_network_convergence(
181-
is_sparse=True, build_strategy=build_strategy, use_cuda=True)
181+
if core.is_compiled_with_cuda():
182+
self.check_network_convergence(
183+
is_sparse=True, build_strategy=build_strategy, use_cuda=True)
182184
self.check_network_convergence(
183185
is_sparse=True, build_strategy=build_strategy, use_cuda=False)
184186

185187
def test_update_dense_parameter_all_reduce(self):
186188
build_strategy = fluid.BuildStrategy()
187189
build_strategy.reduce_strategy = fluid.BuildStrategy.ReduceStrategy.AllReduce
188-
self.check_network_convergence(
189-
is_sparse=False, build_strategy=build_strategy, use_cuda=True)
190+
if core.is_compiled_with_cuda():
191+
self.check_network_convergence(
192+
is_sparse=False, build_strategy=build_strategy, use_cuda=True)
190193
self.check_network_convergence(
191194
is_sparse=False, build_strategy=build_strategy, use_cuda=False)
192195

193196
def test_update_sparse_parameter_reduce(self):
194197
build_strategy = fluid.BuildStrategy()
195198
build_strategy.reduce_strategy = fluid.BuildStrategy.ReduceStrategy.Reduce
196-
self.check_network_convergence(
197-
is_sparse=True, build_strategy=build_strategy, use_cuda=True)
199+
if core.is_compiled_with_cuda():
200+
self.check_network_convergence(
201+
is_sparse=True, build_strategy=build_strategy, use_cuda=True)
198202
self.check_network_convergence(
199203
is_sparse=True, build_strategy=build_strategy, use_cuda=False)
200204

201205
def test_update_dense_parameter_reduce(self):
202206
build_strategy = fluid.BuildStrategy()
203207
build_strategy.reduce_strategy = fluid.BuildStrategy.ReduceStrategy.Reduce
204-
self.check_network_convergence(
205-
is_sparse=False, build_strategy=build_strategy, use_cuda=True)
208+
if core.is_compiled_with_cuda():
209+
self.check_network_convergence(
210+
is_sparse=False, build_strategy=build_strategy, use_cuda=True)
206211
self.check_network_convergence(
207212
is_sparse=False, build_strategy=build_strategy, use_cuda=False)
208213

0 commit comments

Comments
 (0)