rebase

apache · Jan 5, 2020 · d502341 · d502341
1 parent 3673927
commit d502341
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/relay/op/annotation/annotation.cc b/src/relay/op/annotation/annotation.cc
@@ -188,7 +188,7 @@ Beginning of a region that is handled by a given compiler.
                          return {topi::identity(inputs[0])};
                        });
 
-TVM_REGISTER_API("relay.op.annotation._make.compiler_begin")
+TVM_REGISTER_GLOBAL("relay.op.annotation._make.compiler_begin")
 .set_body_typed<Expr(Expr, std::string)>([](Expr expr, std::string compiler) {
   auto attrs = make_object<CompilerAttrs>();
   attrs->compiler = compiler;
@@ -213,7 +213,7 @@ End of a region that is handled by a given compiler.
                          return {topi::identity(inputs[0])};
                        });
 
-TVM_REGISTER_API("relay.op.annotation._make.compiler_end")
+TVM_REGISTER_GLOBAL("relay.op.annotation._make.compiler_end")
 .set_body_typed<Expr(Expr, std::string)>([](Expr expr, std::string compiler) {
   auto attrs = make_object<CompilerAttrs>();
   attrs->compiler = compiler;

diff --git a/src/relay/pass/annotate_compiler.cc b/src/relay/pass/annotate_compiler.cc
@@ -93,7 +93,7 @@ Pass AnnotateCompiler(const std::string& compiler) {
   return transform::Sequential({func_pass, InferType()}, "AnnotateCompiler");
 }
 
-TVM_REGISTER_API("relay._transform.AnnotateCompiler")
+TVM_REGISTER_GLOBAL("relay._transform.AnnotateCompiler")
 .set_body_typed(AnnotateCompiler);
 
 }  // namespace transform

diff --git a/src/relay/pass/partition_graph.cc b/src/relay/pass/partition_graph.cc
@@ -375,7 +375,7 @@ Pass PartitionGraph() {
   return Sequential({partitioned, InferType()});
 }
 
-TVM_REGISTER_API("relay._transform.PartitionGraph")
+TVM_REGISTER_GLOBAL("relay._transform.PartitionGraph")
 .set_body_typed(transform::PartitionGraph);
 
 }  // namespace transform