未验证 提交 c83ade6d 编写于 作者: M mapingshuo 提交者: GitHub

add AsDuplicable for sync_comm op(#27515)

上级 d20349b5
...@@ -54,8 +54,10 @@ class CSyncCommStreamOp : public framework::OperatorBase { ...@@ -54,8 +54,10 @@ class CSyncCommStreamOp : public framework::OperatorBase {
class CSyncCommStreamOpMaker : public framework::OpProtoAndCheckerMaker { class CSyncCommStreamOpMaker : public framework::OpProtoAndCheckerMaker {
public: public:
void Make() { void Make() {
AddInput("X", "(Tensor) Dependency of the variable need to sync"); AddInput("X", "(Tensor) Dependency of the variable need to sync")
AddOutput("Out", "(Tensor) Dependency of the variable need to sync"); .AsDuplicable();
AddOutput("Out", "(Tensor) Dependency of the variable need to sync")
.AsDuplicable();
AddAttr<int>("ring_id", "(int default 0) ring id.").SetDefault(0); AddAttr<int>("ring_id", "(int default 0) ring id.").SetDefault(0);
AddComment(R"DOC( AddComment(R"DOC(
CSyncCommStream Operator CSyncCommStream Operator
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册