mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-09 03:47:57 +00:00
079bf3cb26
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
15 lines
487 B
Python
15 lines
487 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",
|
|
]
|