ColossalAI/colossalai/legacy/zero/shard_utils/__init__.py

from .base_shard_strategy import BaseShardStrategy
from .bucket_tensor_shard_strategy import BucketTensorShardStrategy
from .tensor_shard_strategy import TensorShardStrategy

__all__ = ["BaseShardStrategy", "TensorShardStrategy", "BucketTensorShardStrategy"]
add bucket tensor shard strategy 2022-03-14 06:48:32 +00:00			`from .base_shard_strategy import BaseShardStrategy`
			`from .bucket_tensor_shard_strategy import BucketTensorShardStrategy`
			`from .tensor_shard_strategy import TensorShardStrategy`
[zero] a shard strategy in granularity of tensor (#307) 2022-03-04 03:59:35 +00:00
[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format 2023-09-19 06:20:26 +00:00			`__all__ = ["BaseShardStrategy", "TensorShardStrategy", "BucketTensorShardStrategy"]`