提交 cbfec1f7 编写于 作者: Q qiaolongfei

diable test of sparse_parameter_sgd on GPU

上级 31e8d807
...@@ -157,10 +157,9 @@ class TestSGDOpOptimizeSelectedRows(unittest.TestCase): ...@@ -157,10 +157,9 @@ class TestSGDOpOptimizeSelectedRows(unittest.TestCase):
result_array = np.array(w_tensor) result_array = np.array(w_tensor)
assert (result_array == w_after_optimize).all() assert (result_array == w_after_optimize).all()
def test_sparse_sgd(self): def test_sparse_parameter_sgd(self):
places = [core.CPUPlace()] places = [core.CPUPlace()]
if core.is_compiled_with_cuda(): # do not support GPU kernel currently
places.append(core.CUDAPlace(0))
for place in places: for place in places:
self.check_with_place(place) self.check_with_place(place)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册