[Kernel] Move sgl-kernel under sglang.kernels.aot (#32648)
This commit is contained in:
@@ -840,7 +840,7 @@ FUSION_PATTERN_REGISTRY: Tuple[FusionPatternSpec, ...] = (
|
||||
pattern="PR #22392 CUTLASS FP8 scaled MM replacing nvjet",
|
||||
candidate_path=(
|
||||
"PR #22392"
|
||||
"<br>sgl-kernel/python/sgl_kernel/gemm.py"
|
||||
"<br>python/sglang/kernels/aot/python/sgl_kernel/gemm.py"
|
||||
"<br>python/sglang/srt/layers/quantization/fp8_utils.py"
|
||||
),
|
||||
active_keywords=("cutlass_scaled_mm", "fp8_scaled_mm"),
|
||||
@@ -2632,7 +2632,7 @@ def fusion_framework_hints(spec: FusionPatternSpec) -> set[str]:
|
||||
hints.add("tokenspeed")
|
||||
if "tensorrt_llm/" in text:
|
||||
hints.add("trtllm")
|
||||
if any(token in text for token in ("python/sglang/", "sgl-kernel/", "sgl_kernel/")):
|
||||
if any(token in text for token in ("python/sglang/", "sgl_kernel/")):
|
||||
hints.add("sglang")
|
||||
return hints
|
||||
|
||||
|
||||
Reference in New Issue
Block a user