Skip to content

Commit a33ddf5

Browse files
committed
[microNPU] enable USMP
* rebase fixes -- PoolInfo is moved out of usmp namespace Change-Id: Idcf3b4a9a08d8d430eebee9a748ef0f0878379b0
1 parent b9da4e1 commit a33ddf5

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/tir/usmp/transform/assign_pool_info.cc

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -73,10 +73,10 @@ class PoolInfoAssigner : public StmtExprMutator {
7373
IRModule mod_;
7474
Map<String, Array<PoolInfo>> target_pool_infos_;
7575
PrimFunc func_;
76-
usmp::PoolInfo CreateDefaultMemoryPool(const IRModule& module);
76+
PoolInfo CreateDefaultMemoryPool(const IRModule& module);
7777
};
7878

79-
usmp::PoolInfo PoolInfoAssigner::CreateDefaultMemoryPool(const tvm::IRModule& module) {
79+
PoolInfo PoolInfoAssigner::CreateDefaultMemoryPool(const tvm::IRModule& module) {
8080
Map<Target, String> target_access;
8181
tir::PrimFunc tir_main_func =
8282
Downcast<tir::PrimFunc>(module->Lookup(::tvm::runtime::symbol::tvm_run_func_suffix));
@@ -86,9 +86,9 @@ usmp::PoolInfo PoolInfoAssigner::CreateDefaultMemoryPool(const tvm::IRModule& mo
8686
Optional<Target> target = func->GetAttr<Target>(tvm::attr::kTarget);
8787
target_access.Set(target.value_or(target_host), PoolInfo::kTargetPoolReadWriteAccess);
8888
}
89-
return usmp::PoolInfo("global_workspace", target_access, PoolInfo::kUnrestrictedPoolSizeHint,
90-
PoolInfo::kUnknownClockFrequency, PoolInfo::kUnknownReadBandwidth,
91-
PoolInfo::kUnknownWriteBandwidth, 0, 0, {}, Bool(true));
89+
return PoolInfo("global_workspace", target_access, PoolInfo::kUnrestrictedPoolSizeHint,
90+
PoolInfo::kUnknownClockFrequency, PoolInfo::kUnknownReadBandwidth,
91+
PoolInfo::kUnknownWriteBandwidth, 0, 0, {}, Bool(true));
9292
}
9393

9494
Stmt PoolInfoAssigner::VisitStmt_(const AllocateNode* op) {

0 commit comments

Comments
 (0)