modify pir in distribute api test (#2946)

PaddlePaddle · Sep 12, 2024 · f61a640 · f61a640
1 parent 7d678d6
commit f61a640
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/distributed/CE_API/case/dist_fleet_static.py b/distributed/CE_API/case/dist_fleet_static.py
@@ -62,6 +62,6 @@ def mlp(input_x, input_y, hid_dim=128, label_dim=2):
 step = 5
 train_info = []
 for i in range(step):
-    cost_val = exe.run(program=paddle.static.default_main_program(), feed=gen_data(), fetch_list=[cost.name])
+    cost_val = exe.run(program=paddle.static.default_main_program(), feed=gen_data(), fetch_list=[cost])
     train_info.append(cost_val[0])
 print(train_info)
diff --git a/distributed/CE_API/case/dist_train_launch.py b/distributed/CE_API/case/dist_train_launch.py
@@ -62,6 +62,6 @@ def mlp(input_x, input_y, hid_dim=128, label_dim=2):
 step = 5
 train_info = []
 for i in range(step):
-    cost_val = exe.run(program=paddle.static.default_main_program(), feed=gen_data(), fetch_list=[cost.name])
+    cost_val = exe.run(program=paddle.static.default_main_program(), feed=gen_data(), fetch_list=[cost])
     train_info.append(cost_val[0])
 print(train_info)