mirror of
https://github.com/deepseek-ai/DualPipe
synced 2025-04-29 10:06:58 +00:00
18 lines
278 B
Python
18 lines
278 B
Python
__version__ = "1.0.0"
|
|
|
|
from dualpipe.dualpipe import (
|
|
DualPipe,
|
|
WeightGradStore,
|
|
)
|
|
from dualpipe.comm import (
|
|
set_p2p_tensor_shapes,
|
|
set_p2p_tensor_dtype,
|
|
)
|
|
|
|
__all__ = [
|
|
DualPipe,
|
|
WeightGradStore,
|
|
set_p2p_tensor_shapes,
|
|
set_p2p_tensor_dtype,
|
|
]
|