Skip to content

GitLab

Commit 263b5bde authored by renzhc's avatar renzhc
Browse files

fix compile issue in deepep

parent f810bda3
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
python/sglang/srt/layers/moe/token_dispatcher/deepep.py
View file @ 263b5bde
......@@ -806,6 +806,7 @@ class DeepEPDispatcher(BaseDispatcher):
)
self._dispatch_intermediate_state = inner_state
@torch._dynamo.disable()
def dispatch_b(self):
self._update_stage(_Stage.AFTER_DISPATCH_A, _Stage.AFTER_DISPATCH_B)
inner_state = self._dispatch_intermediate_state
......@@ -833,6 +834,7 @@ class DeepEPDispatcher(BaseDispatcher):
)
self._combine_intermediate_state = inner_state
@torch._dynamo.disable()
def combine_b(self):
self._update_stage(_Stage.AFTER_COMBINE_A, _Stage.INITIAL)
inner_state = self._combine_intermediate_state
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment