[Kernel] Classification cleanup: unify _jit_ naming, drop empty/model groups, add elementwise (RFC #29630) (#32148)
Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.8
parent
1b63155efe
commit
11b0e5c5ad
@@ -453,7 +453,7 @@ FUSION_PATTERN_REGISTRY: Tuple[FusionPatternSpec, ...] = (
|
||||
pattern="In-place QK RMSNorm",
|
||||
candidate_path=(
|
||||
"python/sglang/srt/models/utils.py"
|
||||
"<br>python/sglang/kernels/ops/layernorm/_jit_norm.py"
|
||||
"<br>python/sglang/kernels/ops/layernorm/norm.py"
|
||||
),
|
||||
active_keywords=("fused_inplace_qknorm", "minimaxm2rmsnormtp"),
|
||||
split_groups=(("apply_qk_norm", "q_norm", "k_norm", "qknorm"),),
|
||||
|
||||
Reference in New Issue
Block a user