EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Fix the style of sgl kernel (#10398)

Browse Source
This commit is contained in:
Lianmin Zheng
2025-09-12 22:20:21 -07:00
committed by GitHub
parent 99757cc3e6
commit c9ec4cae5b
5 changed files with 19 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
sgl-kernel/python/sgl_kernel/__init__.py
View File

@@ -34,11 +34,6 @@ from sgl_kernel.elementwise import (
rmsnorm,
silu_and_mul,
)
from sgl_kernel.mamba import causal_conv1d_fwd, causal_conv1d_update
if torch.version.hip is not None:
from sgl_kernel.elementwise import gelu_quick
from sgl_kernel.fused_moe import fused_marlin_moe
from sgl_kernel.gemm import (
awq_dequantize,
@@ -71,6 +66,7 @@ from sgl_kernel.kvcacheio import (
transfer_kv_per_layer,
transfer_kv_per_layer_mla,
)
from sgl_kernel.mamba import causal_conv1d_fwd, causal_conv1d_update
from sgl_kernel.marlin import (
awq_marlin_moe_repack,
awq_marlin_repack,
@@ -104,6 +100,9 @@ from sgl_kernel.speculative import (
from sgl_kernel.top_k import fast_topk
from sgl_kernel.version import __version__
if torch.version.hip is not None:
from sgl_kernel.elementwise import gelu_quick
def create_greenctx_stream_by_value(*args, **kwargs):
from sgl_kernel.spatial import create_greenctx_stream_by_value as _impl