fix UT

2fe35e33 · seiriosPlus · f529675c · 2fe35e33
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

python/paddle/fluid/tests/unittests/test_dist_lookup_sparse_table_fuse_ops.py ...tests/unittests/test_dist_lookup_sparse_table_fuse_ops.py +4 -2

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_dist_lookup_sparse_table_fuse_ops.py
+++ b/python/paddle/fluid/tests/unittests/test_dist_lookup_sparse_table_fuse_ops.py
@@ -94,8 +94,10 @@ class TestLookupTableFuseOp(unittest.TestCase):

        training_program = fluid.Program()

-        scope.var('Beta1Pow').get_tensor().set([0], place)
-        scope.var('Beta2Pow').get_tensor().set([0], place)
+        scope.var('Beta1Pow').get_tensor().set(
+            np.array([0]).astype("float32"), place)
+        scope.var('Beta2Pow').get_tensor().set(
+            np.array([0]).astype("float32"), place)

        rows = [0, 1, 2, 3, 4, 5, 6]
        row_numel = 8