Try to get arm ci container to pass

apache · Jan 19, 2022 · 49ca379 · 49ca379
1 parent 50fdd18
commit 49ca379
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/relay/backend/contrib/cmsisnn/relay_to_tir.cc b/src/relay/backend/contrib/cmsisnn/relay_to_tir.cc
@@ -46,8 +46,10 @@ class RelayToTIRVisitor : public MixedModeMutator {
 
   IRModule Mutate() {
     GlobalVar main_global_var = ir_module_->GetGlobalVar("main");
-    Function main = Downcast<Function>(ir_module_->Lookup(main_global_var));
-    Function mutated_main = WithFields(std::move(main), main->params, VisitExpr(main->body));
+    BaseFunc main = ir_module_->Lookup(main_global_var);
+    Function main_func = GetRef<Function>(main.as<FunctionNode>());
+    Function mutated_main =
+        WithFields(std::move(main_func), main_func->params, VisitExpr(main_func->body));
 
     ir_module_->Update(main_global_var, mutated_main);
 

diff --git a/src/relay/backend/contrib/example_target_hooks/relay_to_tir.cc b/src/relay/backend/contrib/example_target_hooks/relay_to_tir.cc
@@ -43,8 +43,10 @@ class ConvertAddToSubtract : public MixedModeMutator {
 
   IRModule Mutate() {
     GlobalVar main_global_var = ir_module_->GetGlobalVar("main");
-    Function main = GetRef<Function>(ir_module_->Lookup(main_global_var).as<FunctionNode>());
-    Function mutated_main = WithFields(std::move(main), main->params, VisitExpr(main->body));
+    BaseFunc main = ir_module_->Lookup(main_global_var);
+    Function main_func = GetRef<Function>(main.as<FunctionNode>());
+    Function mutated_main =
+        WithFields(std::move(main_func), main_func->params, VisitExpr(main_func->body));
 
     ir_module_->Update(main_global_var, mutated_main);