From b854e0bcd40f84dd705179d0efcae97423c63477 Mon Sep 17 00:00:00 2001
From: Josh Williams <48445365+jvwilliams23@users.noreply.github.com>
Date: Wed, 15 Apr 2026 15:00:19 +0100
Subject: [PATCH] Import forward_halo_exchange to __init__.py

---
 distconv/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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"]