From 49ca37985bce1f43aba8bb6564a7008d7e640e1c Mon Sep 17 00:00:00 2001 From: Lily Orth-Smith Date: Wed, 19 Jan 2022 00:12:36 -0800 Subject: [PATCH] Try to get arm ci container to pass --- src/relay/backend/contrib/cmsisnn/relay_to_tir.cc | 6 ++++-- .../backend/contrib/example_target_hooks/relay_to_tir.cc | 6 ++++-- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/src/relay/backend/contrib/cmsisnn/relay_to_tir.cc b/src/relay/backend/contrib/cmsisnn/relay_to_tir.cc index f9b5b1232a709..152ada3c020bb 100644 --- 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(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(main.as()); + 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 index 6309f4cebcd65..9305927a7caa3 100644 --- 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(ir_module_->Lookup(main_global_var).as()); - Function mutated_main = WithFields(std::move(main), main->params, VisitExpr(main->body)); + BaseFunc main = ir_module_->Lookup(main_global_var); + Function main_func = GetRef(main.as()); + Function mutated_main = + WithFields(std::move(main_func), main_func->params, VisitExpr(main_func->body)); ir_module_->Update(main_global_var, mutated_main);