提交 7a7d27b3 编写于 作者: T typhoonzero

update op

上级 0ae726f0
......@@ -102,8 +102,7 @@ class GenNCCLIdOp : public framework::OperatorBase {
class GenNCCLIdOpMaker : public framework::OpProtoAndCheckerMaker {
public:
GenNCCLIdOpMaker(OpProto* proto, OpAttrChecker* op_checker)
: OpProtoAndCheckerMaker(proto, op_checker) {
void Make() override {
AddOutput("NCCLID", "Raw variable contains a NCCL UniqueId instaces.");
AddComment(R"DOC(
GenNCCLId operator
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册