Skip to content

Commit c26799b

Browse files
committed
modify dtype in doublegrad matmul ut
1 parent 6335e6a commit c26799b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

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

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -161,10 +161,10 @@ def func(self, place):
161161
y_shapes = [[2], [3, 2], [2, 4, 5], [2, 3, 3, 5], [4, 3]]
162162
transpose_xs = [False, True, True, False, False]
163163
transpose_ys = [False, True, False, True, False]
164-
dtypes = [np.float64, np.float64, np.float32, np.float32, np.float64]
165-
typenames = ["float64", "float64", "float32", "float32", "float64"]
166-
for i, (x_shape, y_shape, transpose_x, transpose_y, dtype, typename) \
167-
in enumerate(zip(x_shapes, y_shapes, transpose_xs, transpose_ys, dtypes, typenames)):
164+
dtype = np.float64
165+
typename = "float64"
166+
for i, (x_shape, y_shape, transpose_x, transpose_y) \
167+
in enumerate(zip(x_shapes, y_shapes, transpose_xs, transpose_ys)):
168168
x = layers.create_parameter(
169169
dtype=typename, shape=x_shape, name='x{}'.format(i))
170170
y = layers.create_parameter(

0 commit comments

Comments
 (0)