deep_gemm/utils/__init__.py

from . import math, layout
from .layout import *
from .math import *
from .dist import init_dist, uneven_all_gather
Add more GPU architectures support (#112) * Add more GPU architectures support * Update layout.py * Optimize performance, Add SM90 support, Add 1D2D SM100 support * Add fmtlib submodule at commit 553ec11 --------- Co-authored-by: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com> 2025-07-18 11:32:22 +08:00			`from . import math, layout`
			`from .layout import *`
			`from .math import *`
[Public release 26/04] Introducing Mega MoE, FP4 Indexer and other features/fixes (#304) * Merge with private repo * Update README * Update README * Update README * Add PyTorch requirements * Fix sync scopes for MQA logits (#256) * Update README 2026-04-17 09:45:14 +08:00			`from .dist import init_dist, uneven_all_gather`