github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

remove personal comments

pull/5733/head
genghaozhe 2024-05-20 07:28:20 +00:00
parent bfcb2d1ff8
commit 90d8d0183c
1 changed files with 0 additions and 1 deletions
Show Stats 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() self._gemini_manager.sample_overall_data()
# evit chunks, aware of async fetched # evit chunks, aware of async fetched
# TODO: check if prefetched chunks will be evicted
self._gemini_manager.adjust_layout( self._gemini_manager.adjust_layout(
all_chunks, record_anyway=self._gemini_manager.placement_policy.max_prefetch > 0 all_chunks, record_anyway=self._gemini_manager.placement_policy.max_prefetch > 0
) )