From 8547562884c4bfec614059ec5ae775a326d78036 Mon Sep 17 00:00:00 2001 From: hxwang Date: Tue, 28 May 2024 05:16:02 +0000 Subject: [PATCH] [chore] remove unnecessary assert since compute list might not be recorded --- colossalai/zero/gemini/gemini_mgr.py | 1 - 1 file changed, 1 deletion(-) diff --git a/colossalai/zero/gemini/gemini_mgr.py b/colossalai/zero/gemini/gemini_mgr.py index d6b539f55..83e475575 100644 --- 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