From 1f7cee81da9aa8c68610a947b1e622add9217992 Mon Sep 17 00:00:00 2001
From: Yilong Zhao <74357408+happierpig@users.noreply.github.com>
Date: Tue, 31 Mar 2026 16:32:47 -0700
Subject: [PATCH] [moe] add customized option to moe-a2a-backend (#21786)

---
 python/sglang/srt/layers/moe/utils.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/python/sglang/srt/layers/moe/utils.py b/python/sglang/srt/layers/moe/utils.py
index 65da83c0e..0d5fa7ddb 100644
--- a/python/sglang/srt/layers/moe/utils.py
+++ b/python/sglang/srt/layers/moe/utils.py
@@ -26,6 +26,7 @@ class MoeA2ABackend(Enum):
     MORI = "mori"
     ASCEND_FUSEEP = "ascend_fuseep"
     FLASHINFER = "flashinfer"
+    CUSTOMIZED = "customized"
 
     @classmethod
     def _missing_(cls, value):
@@ -57,6 +58,9 @@ class MoeA2ABackend(Enum):
     def is_mori(self):
         return self == MoeA2ABackend.MORI
 
+    def is_customized(self):
+        return self == MoeA2ABackend.CUSTOMIZED
+
 
 class MoeRunnerBackend(Enum):