[chore] remove unnecessary assert since compute list might not be recorded

2024-05-28 05:16:02 +00:00 · 2024-05-28 05:16:02 +00:00 · 8547562884
parent e5e3320948
commit 8547562884
1 changed files with 0 additions and 1 deletions
--- a/colossalai/zero/gemini/gemini_mgr.py
+++ b/colossalai/zero/gemini/gemini_mgr.py
@ -57,7 +57,6 @@ class GeminiManager:
        self._comp_cuda_demand_time = 0

    def reset_attributes(self):
-        assert self._compute_idx + 1 == len(self._compute_list)
        self._compute_idx = -1
        self._h2d_volume = 0
        self._d2h_volume = 0