github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-07 12:01:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove perf log, unrelated file and so on

Browse Source
This commit is contained in:
genghaozhe
2024-05-20 05:21:26 +00:00
parent 5c6c5d6be3
commit 1ec92d29af
7 changed files with 7 additions and 151 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/zero/gemini/gemini_mgr.py
View File

@@ -125,7 +125,7 @@ class GeminiManager:
self._async_works[chunk].wait()
del self._async_works[chunk]
else:
non_prefetched_chunks.append(chunk) # 没在之前prefetch过，现在要prefetch的chunk
non_prefetched_chunks.append(chunk)
return tuple(non_prefetched_chunks)
def add_work(self, chunk: Chunk, work: dist.Work):