Update include/tvm/relay/attrs/annotation.h

Co-Authored-By: 雾雨魔理沙 <lolisa@marisa.moe>
apache · Jan 1, 2020 · 23154bd · 23154bd
1 parent 3228135
commit 23154bd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/include/tvm/relay/attrs/annotation.h b/include/tvm/relay/attrs/annotation.h
@@ -66,7 +66,7 @@ struct CompilerAttrs : public tvm::AttrsNode<CompilerAttrs> {
 
   TVM_DECLARE_ATTRS(CompilerAttrs, "relay.attrs.CompilerAttrs") {
     TVM_ATTR_FIELD(compiler)
-      .describe("A 3rd compiler used for code generation.");
+      .describe("A 3rd party compiler used for code generation.");
   }
 };
 

diff --git a/src/relay/op/annotation/annotation.cc b/src/relay/op/annotation/annotation.cc
@@ -190,7 +190,7 @@ Beginning of a region that is handled by a given compiler.
 
 TVM_REGISTER_API("relay.op.annotation._make.compiler_begin")
 .set_body_typed<Expr(Expr, std::string)>([](Expr expr, std::string compiler) {
-  auto attrs = make_node<CompilerAttrs>();
+  auto attrs = make_object<CompilerAttrs>();
   attrs->compiler = compiler;
   static const Op& op = Op::Get("annotation.compiler_begin");
   return CallNode::make(op, {expr}, Attrs(attrs), {});
@@ -215,7 +215,7 @@ End of a region that is handled by a given compiler.
 
 TVM_REGISTER_API("relay.op.annotation._make.compiler_end")
 .set_body_typed<Expr(Expr, std::string)>([](Expr expr, std::string compiler) {
-  auto attrs = make_node<CompilerAttrs>();
+  auto attrs = make_object<CompilerAttrs>();
   attrs->compiler = compiler;
   static const Op& op = Op::Get("annotation.compiler_end");
   return CallNode::make(op, {expr}, Attrs(attrs), {});