From 9daafb126dde695213c833b9358b077abc443d7d Mon Sep 17 00:00:00 2001
From: Manupa Karunaratne <manupa.karunaratne@arm.com>
Date: Mon, 14 Feb 2022 14:26:16 +0000
Subject: [PATCH] [microNPU] enable USMP

* rebase fixes -- PoolInfo is moved out of usmp namespace

Change-Id: Idcf3b4a9a08d8d430eebee9a748ef0f0878379b0
---
 src/tir/usmp/transform/assign_pool_info.cc | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/tir/usmp/transform/assign_pool_info.cc b/src/tir/usmp/transform/assign_pool_info.cc
index e1f0114e75e9..9df4cdda31db 100644
--- a/src/tir/usmp/transform/assign_pool_info.cc
+++ b/src/tir/usmp/transform/assign_pool_info.cc
@@ -73,10 +73,10 @@ class PoolInfoAssigner : public StmtExprMutator {
   IRModule mod_;
   Map<String, Array<PoolInfo>> target_pool_infos_;
   PrimFunc func_;
-  usmp::PoolInfo CreateDefaultMemoryPool(const IRModule& module);
+  PoolInfo CreateDefaultMemoryPool(const IRModule& module);
 };
 
-usmp::PoolInfo PoolInfoAssigner::CreateDefaultMemoryPool(const tvm::IRModule& module) {
+PoolInfo PoolInfoAssigner::CreateDefaultMemoryPool(const tvm::IRModule& module) {
   Map<Target, String> target_access;
   tir::PrimFunc tir_main_func =
       Downcast<tir::PrimFunc>(module->Lookup(::tvm::runtime::symbol::tvm_run_func_suffix));
@@ -86,9 +86,9 @@ usmp::PoolInfo PoolInfoAssigner::CreateDefaultMemoryPool(const tvm::IRModule& mo
     Optional<Target> target = func->GetAttr<Target>(tvm::attr::kTarget);
     target_access.Set(target.value_or(target_host), PoolInfo::kTargetPoolReadWriteAccess);
   }
-  return usmp::PoolInfo("global_workspace", target_access, kUnrestrictedPoolSizeHint,
-                        kUnknownClockFrequency, kUnknownReadBandwidth,
-                        kUnknownWriteBandwidth, 0, 0, {}, Bool(true));
+  return PoolInfo("global_workspace", target_access, kUnrestrictedPoolSizeHint,
+                  kUnknownClockFrequency, kUnknownReadBandwidth,
+                  kUnknownWriteBandwidth, 0, 0, {}, Bool(true));
 }
 
 Stmt PoolInfoAssigner::VisitStmt_(const AllocateNode* op) {