[hotfix]: add memory utils to avoid import legacy module#5123

cwher merged 1 commit into

hpcaitech:feature/shardformerhpcaitech/ColossalAI:feature/shardformerfrom

cwher:feature/shardformercwher/ColossalAI:feature/shardformerCopy head branch name to clipboard