From cbfec1f7d614835b806e0f37adb1cfcd8b15b444 Mon Sep 17 00:00:00 2001 From: qiaolongfei Date: Tue, 3 Apr 2018 19:56:19 +0800 Subject: [PATCH] diable test of sparse_parameter_sgd on GPU --- python/paddle/fluid/tests/unittests/test_sgd_op.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_sgd_op.py b/python/paddle/fluid/tests/unittests/test_sgd_op.py index 4761b435480..3126293f9d8 100644 --- a/python/paddle/fluid/tests/unittests/test_sgd_op.py +++ b/python/paddle/fluid/tests/unittests/test_sgd_op.py @@ -157,10 +157,9 @@ class TestSGDOpOptimizeSelectedRows(unittest.TestCase): result_array = np.array(w_tensor) assert (result_array == w_after_optimize).all() - def test_sparse_sgd(self): + def test_sparse_parameter_sgd(self): places = [core.CPUPlace()] - if core.is_compiled_with_cuda(): - places.append(core.CUDAPlace(0)) + # do not support GPU kernel currently for place in places: self.check_with_place(place) -- GitLab