DeepSeek/DualPipe
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DualPipe synced 2025-06-26 18:16:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add dualpipev

Browse Source
This commit is contained in:
ljss
2025-03-04 17:50:00 +08:00
parent 036c14e7f6
commit 8ec5883b30
8 changed files with 609 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
dualpipe/__init__.py
View File

@@ -1,16 +1,16 @@
__version__ = "1.0.0"
from dualpipe.dualpipe import (
DualPipe,
WeightGradStore,
)
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,