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

[compatibility] ChunkMgr import error (#1772)

pull/1778/head
Jiarui Fang 2 years ago committed by GitHub
parent
a4d1f59c78
commit
f34dab4270
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 8
      colossalai/gemini/__init__.py

8
colossalai/gemini/__init__.py
Unescape View File

@ -1,6 +1,8 @@
from .chunk import TensorInfo, TensorState
from .chunk import ChunkManager, TensorInfo, TensorState
from .gemini_mgr import GeminiManager
from .stateful_tensor_mgr import StatefulTensorMgr
from .tensor_placement_policy import TensorPlacementPolicyFactory
from .gemini_mgr import GeminiManager
__all__ = ['StatefulTensorMgr', 'TensorPlacementPolicyFactory', 'GeminiManager', 'TensorInfo', 'TensorState']
__all__ = [
'StatefulTensorMgr', 'TensorPlacementPolicyFactory', 'GeminiManager', 'TensorInfo', 'TensorState', 'ChunkManager'
]

Write Preview
Loading…
Cancel
Save