From 31dddd0f5b008095f26c21c1ea547d4005cf2489 Mon Sep 17 00:00:00 2001 From: Egor Churaev Date: Mon, 12 Dec 2022 14:21:38 +0300 Subject: [PATCH] [Adreno] Update interface of AnnotateMemoryScope pass Removed redundant and unused parameter from AnnotateMemoryScope pass. --- include/tvm/relay/transform.h | 2 +- src/relay/backend/build_module.cc | 2 +- src/relay/transforms/annotate_texture_storage.cc | 8 ++++---- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/include/tvm/relay/transform.h b/include/tvm/relay/transform.h index 3227f7979d87..43a0f89d95c1 100644 --- a/include/tvm/relay/transform.h +++ b/include/tvm/relay/transform.h @@ -584,7 +584,7 @@ TVM_DLL Pass CapturePostDfsIndexInSpans(); * \brief Calls device dependent memory scope analysis pass, collects mapping of desirable * expr->memory_scope and annotates expressions by VirtualDevice with required memory_scope */ -TVM_DLL Pass AnnotateMemoryScope(CompilationConfig config); +TVM_DLL Pass AnnotateMemoryScope(); /*! * \brief Removes non-fused reshapes after lowering the graph. diff --git a/src/relay/backend/build_module.cc b/src/relay/backend/build_module.cc index bca524794a20..0642c0c67253 100644 --- a/src/relay/backend/build_module.cc +++ b/src/relay/backend/build_module.cc @@ -396,7 +396,7 @@ class RelayBuildModule : public runtime::ModuleNode { relay_module = transform::Inline()(relay_module); relay_module = transform::InferType()(relay_module); relay_module = transform::LabelOps()(relay_module); - relay_module = transform::AnnotateMemoryScope(config_)(relay_module); + relay_module = transform::AnnotateMemoryScope()(relay_module); ICHECK(relay_module.defined()); diff --git a/src/relay/transforms/annotate_texture_storage.cc b/src/relay/transforms/annotate_texture_storage.cc index 9dbd631ad32d..39f065ea8c11 100644 --- a/src/relay/transforms/annotate_texture_storage.cc +++ b/src/relay/transforms/annotate_texture_storage.cc @@ -645,7 +645,7 @@ Map>> CollectStorageInfo(const Expr& expr) { return storage_info; } -Expr AnnotateMemoryScopeExpr(const Expr& expr, const IRModule& mod, CompilationConfig config) { +Expr AnnotateMemoryScopeExpr(const Expr& expr, const IRModule& mod) { auto storage_scope = CollectStorageInfo(expr); if (storage_scope.size()) { return RewriteVDStorageScopes(storage_scope).Rewrite(expr); @@ -655,10 +655,10 @@ Expr AnnotateMemoryScopeExpr(const Expr& expr, const IRModule& mod, CompilationC } namespace transform { -tvm::transform::Pass AnnotateMemoryScope(CompilationConfig config) { +tvm::transform::Pass AnnotateMemoryScope() { runtime::TypedPackedFunc pass_func = - [config = std::move(config)](Function f, IRModule m, PassContext pc) { - return Downcast(AnnotateMemoryScopeExpr(f, m, config)); + [](Function f, IRModule m, PassContext pc) { + return Downcast(AnnotateMemoryScopeExpr(f, m)); }; return CreateFunctionPass(pass_func, 2, "AnnotateMemoryScope", {}); }