[Doc] Add developer guide for CustomOp (#30886)

Signed-off-by: shen-shanshan <467638484@qq.com>
2026-01-10 00:21:11 +08:00
parent ac9f9330e6
commit 08d954f036
24 changed files with 441 additions and 5 deletions
--- a/vllm/model_executor/layers/mamba/mamba_mixer.py
+++ b/vllm/model_executor/layers/mamba/mamba_mixer.py
@@ -39,6 +39,7 @@ from vllm.v1.attention.backends.mamba1_attn import Mamba1AttentionMetadata


 # Adapted from transformers.models.mamba.modeling_mamba.MambaMixer
+# --8<-- [start:mamba_mixer]
@CustomOp.register("mamba_mixer")
 class MambaMixer(MambaBase, CustomOp):
    """
@@ -51,6 +52,8 @@ class MambaMixer(MambaBase, CustomOp):
    **selective** state spaces)
    """

+    # --8<-- [end:mamba_mixer]
+
    def __init__(
        self,
        hidden_size: int,