Skip to content

Commit 6bd2d2b

Browse files
authored
[Phi] move the graph_send_recv op to the phi (#40092)
* [Phi] transfer old kernel to pten kernel for the graph_send_recv op * update the code for the define of graph_send_recv * fix the gradient problem for graph_send_recv * fix the compile problem * update the enfore message for the windows * update the code for the compiler * update compiler problem for the windows * udpate the code for windows * fix some format problem
1 parent 413a743 commit 6bd2d2b

12 files changed

+999
-721
lines changed

paddle/fluid/operators/graph_send_recv_op.cc

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
See the License for the specific language governing permissions and
1313
limitations under the License. */
1414

15-
#include "paddle/fluid/operators/graph_send_recv_op.h"
15+
#include "paddle/fluid/framework/op_registry.h"
1616

1717
namespace paddle {
1818
namespace operators {
@@ -171,13 +171,3 @@ REGISTER_OPERATOR(graph_send_recv, ops::GraphSendRecvOP,
171171
ops::GraphSendRecvGradOpMaker<paddle::framework::OpDesc>,
172172
ops::GraphSendRecvGradOpMaker<paddle::imperative::OpBase>);
173173
REGISTER_OPERATOR(graph_send_recv_grad, ops::GraphSendRecvGradOp);
174-
REGISTER_OP_CPU_KERNEL(graph_send_recv, ops::GraphSendRecvOpKernel<CPU, float>,
175-
ops::GraphSendRecvOpKernel<CPU, double>,
176-
ops::GraphSendRecvOpKernel<CPU, int>,
177-
ops::GraphSendRecvOpKernel<CPU, int64_t>);
178-
179-
REGISTER_OP_CPU_KERNEL(graph_send_recv_grad,
180-
ops::GraphSendRecvGradOpKernel<CPU, float>,
181-
ops::GraphSendRecvGradOpKernel<CPU, double>,
182-
ops::GraphSendRecvGradOpKernel<CPU, int>,
183-
ops::GraphSendRecvGradOpKernel<CPU, int64_t>);

0 commit comments

Comments
 (0)