未验证 提交 4670a0a2 编写于 作者: L Leo Chen 提交者: GitHub

fix test_fc_op (#33417)

上级 64914ea4
...@@ -138,6 +138,7 @@ class TestFcOp_NumFlattenDims_NegOne(unittest.TestCase): ...@@ -138,6 +138,7 @@ class TestFcOp_NumFlattenDims_NegOne(unittest.TestCase):
def test_api(self): def test_api(self):
def run_program(num_flatten_dims): def run_program(num_flatten_dims):
paddle.seed(SEED) paddle.seed(SEED)
np.random.seed(SEED)
startup_program = Program() startup_program = Program()
main_program = Program() main_program = Program()
...@@ -158,6 +159,7 @@ class TestFcOp_NumFlattenDims_NegOne(unittest.TestCase): ...@@ -158,6 +159,7 @@ class TestFcOp_NumFlattenDims_NegOne(unittest.TestCase):
exe = fluid.Executor(place=place) exe = fluid.Executor(place=place)
exe.run(startup_program) exe.run(startup_program)
out = exe.run(main_program, feed={"x": input}, fetch_list=[out]) out = exe.run(main_program, feed={"x": input}, fetch_list=[out])
return out
res_1 = run_program(-1) res_1 = run_program(-1)
res_2 = run_program(2) res_2 = run_program(2)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册