rename to task extraction

masahi · Mar 11, 2022 · 3c5a318 · 3c5a318
1 parent 57f2882
commit 3c5a318
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...y/backend/metaschedule_task_extraction.cc → src/relay/backend/task_extraction.cc b/...y/backend/metaschedule_task_extraction.cc → src/relay/backend/task_extraction.cc
@@ -30,6 +30,7 @@
 namespace tvm {
 namespace relay {
 namespace backend {
+
 namespace metaschedule {
 
 using meta_schedule::ExtractedTask;
@@ -63,12 +64,13 @@ Array<ExtractedTask> ExtractTask(IRModule mod, Target target, Map<String, Consta
   return tasks;
 }
 
+}  // namespace metaschedule
+
 TVM_REGISTER_GLOBAL("relay.backend.MetaScheduleExtractTask")
     .set_body_typed([](IRModule mod, Target target, Map<String, Constant> params) {
-      return ExtractTask(mod, target, params);
+      return metaschedule::ExtractTask(mod, target, params);
     });
 
-}  // namespace metaschedule
 }  // namespace backend
 }  // namespace relay
 }  // namespace tvm