Update BMM_Theory: bmm_theory/branches/__init__.py

2026-09-03 09:25:39 +00:00
parent 5bee731570
commit 113769b657
1 changed files with 8 additions and 1 deletions
--- 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",
+]