@@ -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
9494Stmt PoolInfoAssigner::VisitStmt_ (const AllocateNode* op) {
0 commit comments