PaddlePaddle
diff --git a/‎python/paddle/distributed/__init__.py
Copy file name to clipboard
+6Lines changed: 6 additions & 0 deletions b/‎python/paddle/distributed/__init__.py
Copy file name to clipboard
+6Lines changed: 6 additions & 0 deletions
diff --git a/‎python/paddle/distributed/auto_parallel/__init__.py
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions b/‎python/paddle/distributed/auto_parallel/__init__.py
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions
@@ -48,6 +48,10 @@
 )
 from .auto_parallel.high_level_api import to_distributed
 from .auto_parallel.interface import get_mesh, set_mesh
+from .auto_parallel.intermediate.context_parallel import (
+    ContextParallel,
+    PrepareContextParallel,
+)
 from .auto_parallel.intermediate.parallelize import parallelize
 from .auto_parallel.intermediate.pipeline_parallel import SplitPoint
 from .auto_parallel.intermediate.tensor_parallel import (
@@ -219,4 +223,6 @@
     "get_mesh",
     "to_distributed",
     "ConvParallel",
+    "ContextParallel",
+    "PrepareContextParallel",
 ]
@@ -24,6 +24,7 @@
 )
 from .process_mesh import ProcessMesh  # noqa: F401
 from .random import parallel_manual_seed  # noqa: F401
+from .ring_attention import RingFlashAttention  # noqa: F401
 from .ring_conv import RingConv2d  # noqa: F401
 from .static.engine import Engine  # noqa: F401
 from .strategy import Strategy  # noqa: F401
Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,7 @@`
`24`	`24`	`)`
`25`	`25`	`from .process_mesh import ProcessMesh # noqa: F401`
`26`	`26`	`from .random import parallel_manual_seed # noqa: F401`
	`27`	`+from .ring_attention import RingFlashAttention # noqa: F401`
`27`	`28`	`from .ring_conv import RingConv2d # noqa: F401`
`28`	`29`	`from .static.engine import Engine # noqa: F401`
`29`	`30`	`from .strategy import Strategy # noqa: F401`