mirror of
https://github.com/deepseek-ai/DualPipe
synced 2025-04-18 13:15:14 +00:00
18 lines
347 B
Python
18 lines
347 B
Python
__version__ = "1.0.0"
|
|
|
|
from dualpipe.dualpipe import DualPipe
|
|
from dualpipe.dualpipev import DualPipeV
|
|
from dualpipe.comm import (
|
|
set_p2p_tensor_shapes,
|
|
set_p2p_tensor_dtype,
|
|
)
|
|
from dualpipe.utils import WeightGradStore
|
|
|
|
__all__ = [
|
|
DualPipe,
|
|
DualPipeV,
|
|
WeightGradStore,
|
|
set_p2p_tensor_shapes,
|
|
set_p2p_tensor_dtype,
|
|
]
|