diff --git a/python/paddle/fluid/tests/unittests/test_sgd_op.py b/python/paddle/fluid/tests/unittests/test_sgd_op.py index 4761b43548074b2cd55b2014ac4ac3898d6ff9e4..3126293f9d8e52daa866be5fc1533648a33f3363 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)