Skip to content

Commit cbfec1f

Browse files
committed
diable test of sparse_parameter_sgd on GPU
1 parent 31e8d80 commit cbfec1f

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

python/paddle/fluid/tests/unittests/test_sgd_op.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -157,10 +157,9 @@ def check_with_place(self, place):
157157
result_array = np.array(w_tensor)
158158
assert (result_array == w_after_optimize).all()
159159

160-
def test_sparse_sgd(self):
160+
def test_sparse_parameter_sgd(self):
161161
places = [core.CPUPlace()]
162-
if core.is_compiled_with_cuda():
163-
places.append(core.CUDAPlace(0))
162+
# do not support GPU kernel currently
164163
for place in places:
165164
self.check_with_place(place)
166165

0 commit comments

Comments
 (0)