github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-13 21:22:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

[polish] add update directory in gemini; rename AgChunk to ChunkV2 (#1432)

Browse Source
This commit is contained in:
HELSON
2022-08-10 16:40:29 +08:00
committed by GitHub
parent f20cb4e893
commit 039b7ed3bc
3 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/gemini/update/__init__.py Normal file
View File

@@ -0,0 +1 @@
from .chunkv2 import ChunkV2

2
colossalai/gemini/ag_chunk.py → colossalai/gemini/update/chunkv2.py
View File

@@ -8,7 +8,7 @@ from colossalai.gemini.chunk import TensorState, STATE_TRANS, TensorInfo, ChunkF
free_storage, alloc_storage
class AgChunk:
class ChunkV2:
def __init__(self,
chunk_size: int,
process_group: ColoProcessGroup,