From 013690a86b6906d0b3eaa11173a9510856be09b3 Mon Sep 17 00:00:00 2001
From: genghaozhe <939857490@qq.com>
Date: Thu, 16 May 2024 09:57:51 +0000
Subject: [PATCH] remove set(all_chunks)

---
 colossalai/zero/gemini/gemini_hook.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/colossalai/zero/gemini/gemini_hook.py b/colossalai/zero/gemini/gemini_hook.py
index d1fd0867f..27a19c132 100644
--- a/colossalai/zero/gemini/gemini_hook.py
+++ b/colossalai/zero/gemini/gemini_hook.py
@@ -33,7 +33,6 @@ class GeminiZeROHook(ColoParamOpHook):
         all_chunks = self._chunk_manager.get_chunks(params)
 
         # wait for prefetched chunks, filter those are not prefetched
-        set(all_chunks)
         chunks_fetch_sync = self._gemini_manager.wait_chunks(all_chunks)
 
         # transfer state