|
16 | 16 |
|
17 | 17 | import paddle.dataset.conll05 as conll05
|
18 | 18 | import paddle.fluid as fluid
|
| 19 | +import paddle.fluid.core as core |
19 | 20 | import unittest
|
20 | 21 | import paddle
|
21 | 22 | import numpy as np
|
@@ -177,32 +178,36 @@ def check_network_convergence(self,
|
177 | 178 | def test_update_sparse_parameter_all_reduce(self):
|
178 | 179 | build_strategy = fluid.BuildStrategy()
|
179 | 180 | 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) |
182 | 184 | self.check_network_convergence(
|
183 | 185 | is_sparse=True, build_strategy=build_strategy, use_cuda=False)
|
184 | 186 |
|
185 | 187 | def test_update_dense_parameter_all_reduce(self):
|
186 | 188 | build_strategy = fluid.BuildStrategy()
|
187 | 189 | 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) |
190 | 193 | self.check_network_convergence(
|
191 | 194 | is_sparse=False, build_strategy=build_strategy, use_cuda=False)
|
192 | 195 |
|
193 | 196 | def test_update_sparse_parameter_reduce(self):
|
194 | 197 | build_strategy = fluid.BuildStrategy()
|
195 | 198 | 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) |
198 | 202 | self.check_network_convergence(
|
199 | 203 | is_sparse=True, build_strategy=build_strategy, use_cuda=False)
|
200 | 204 |
|
201 | 205 | def test_update_dense_parameter_reduce(self):
|
202 | 206 | build_strategy = fluid.BuildStrategy()
|
203 | 207 | 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) |
206 | 211 | self.check_network_convergence(
|
207 | 212 | is_sparse=False, build_strategy=build_strategy, use_cuda=False)
|
208 | 213 |
|
|
0 commit comments