mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-13 05:35:38 +00:00
26b7aac0be
* [zero] refactor low-level zero folder structure * [zero] fix legacy zero import path * [zero] fix legacy zero import path * [zero] remove useless import * [zero] refactor gemini folder structure * [zero] refactor gemini folder structure * [zero] refactor legacy zero import path * [zero] refactor gemini folder structure * [zero] refactor gemini folder structure * [zero] refactor gemini folder structure * [zero] refactor legacy zero import path * [zero] fix test import path * [zero] fix test * [zero] fix circular import * [zero] update import
10 lines
466 B
Python
10 lines
466 B
Python
from .ophooks import BaseOpHook, register_ophooks_recursively
|
|
from .stateful_tensor import StatefulTensor
|
|
from .stateful_tensor_mgr import StatefulTensorMgr
|
|
from .tensor_placement_policy import AutoTensorPlacementPolicy, CPUTensorPlacementPolicy, CUDATensorPlacementPolicy
|
|
|
|
__all__ = [
|
|
'StatefulTensorMgr', 'StatefulTensor', 'CPUTensorPlacementPolicy', 'CUDATensorPlacementPolicy',
|
|
'AutoTensorPlacementPolicy', 'register_ophooks_recursively', 'BaseOpHook'
|
|
]
|