From 113769b657614ad71f0de414e00e53016a895649 Mon Sep 17 00:00:00 2001
From: admin <zhyaction@foxmail.com>
Date: Thu, 3 Sep 2026 09:25:39 +0000
Subject: [PATCH] Update BMM_Theory: bmm_theory/branches/__init__.py

---
 BMM/BMM_Theory/bmm_theory/branches/__init__.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/BMM/BMM_Theory/bmm_theory/branches/__init__.py b/BMM/BMM_Theory/bmm_theory/branches/__init__.py
index 2c955b2..3894e03 100644
--- a/BMM/BMM_Theory/bmm_theory/branches/__init__.py
+++ b/BMM/BMM_Theory/bmm_theory/branches/__init__.py
@@ -1,4 +1,11 @@
 from .merge_batch import MergeBatchBranch
 from .iter_batch import IterBatchBranch
+from .to_matmul import ToMatmulBranch
+from .special import SpecialBranch
+from .stream_k import StreamKBranch
+from .asw_basic import AswBasicBranch
 
-__all__ = ["MergeBatchBranch", "IterBatchBranch"]
+__all__ = [
+    "MergeBatchBranch", "IterBatchBranch", "ToMatmulBranch",
+    "SpecialBranch", "StreamKBranch", "AswBasicBranch",
+]