提交 1a4f79a7 编写于 作者: Y Yancey1989

fix unittest test=develop

上级 a8612adb
...@@ -153,6 +153,7 @@ std::unique_ptr<ir::Graph> BuildStrategy::Apply( ...@@ -153,6 +153,7 @@ std::unique_ptr<ir::Graph> BuildStrategy::Apply(
pass->Erase("local_scopes"); pass->Erase("local_scopes");
pass->SetNotOwned<const std::vector<Scope *>>("local_scopes", pass->SetNotOwned<const std::vector<Scope *>>("local_scopes",
&local_scopes); &local_scopes);
pass->Erase("num_parallel_devices");
pass->Set<size_t>("num_parallel_devices", pass->Set<size_t>("num_parallel_devices",
new size_t(num_parallel_devices)); new size_t(num_parallel_devices));
......
...@@ -211,7 +211,9 @@ class TestCRFModel(unittest.TestCase): ...@@ -211,7 +211,9 @@ class TestCRFModel(unittest.TestCase):
use_cuda=True) use_cuda=True)
self.check_network_convergence( self.check_network_convergence(
is_sparse=False, build_strategy=build_strategy, use_cuda=False) is_sparse=False,
build_strategy=self._new_build_strategy(),
use_cuda=False)
def test_update_sparse_parameter_reduce(self): def test_update_sparse_parameter_reduce(self):
if core.is_compiled_with_cuda(): if core.is_compiled_with_cuda():
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册