change api attribute name: pool_type to reduce_op, compute_type to me…

…ssage_op
PaddlePaddle · Aug 11, 2022 · e7cbc9f · e7cbc9f
1 parent 50bf7da
commit e7cbc9f
Show file tree

Hide file tree

Showing 3 changed files with 94 additions and 97 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_graph_send_recv_op.py b/python/paddle/fluid/tests/unittests/test_graph_send_recv_op.py
@@ -25,11 +25,11 @@
 def graph_send_recv_wrapper(x,
                             src_index,
                             dst_index,
-                            pool_type="sum",
+                            reduce_op="sum",
                             out_size=None,
                             name=None):
     return paddle.geometric.send_u_recv(x, src_index, dst_index,
-                                        pool_type.lower(), out_size, name)
+                                        reduce_op.lower(), out_size, name)
 
 
 class TestGraphSendRecvMaxOp(OpTest):