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

remove personal comments

Browse Source
This commit is contained in:
genghaozhe
2024-05-20 07:28:20 +00:00
parent bfcb2d1ff8
commit 90d8d0183c
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/zero/gemini/gemini_hook.py
View File

@@ -41,7 +41,6 @@ class GeminiZeROHook(ColoParamOpHook):
self._gemini_manager.sample_overall_data()
# evit chunks, aware of async fetched
# TODO: check if prefetched chunks will be evicted
self._gemini_manager.adjust_layout(
all_chunks, record_anyway=self._gemini_manager.placement_policy.max_prefetch > 0
)