diff --git a/distconv/__init__.py b/distconv/__init__.py
index 5b1d0c9..e19fb3a 100644
--- a/distconv/__init__.py
+++ b/distconv/__init__.py
@@ -1,4 +1,4 @@
 from .ddp import DistConvDDP
-from .distconv import DCTensor, ParallelStrategy
+from .distconv import DCTensor, ParallelStrategy, forward_halo_exchange
 
 __all__ = ["DistConvDDP", "DCTensor", "ParallelStrategy"]