Skip to content

GitLab

Unverified Commit 76c48a09 authored by Cheng Wan's avatar Cheng Wan Committed by GitHub
Browse files

[DeepEP] fix: import buffer error (#5179)

parent 90caf06c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/sglang/srt/layers/moe/ep_moe/token_dispatcher.py
View file @ 76c48a09
...@@ -28,7 +28,7 @@ class DeepEPDispatchMode(IntEnum): ...@@ -28,7 +28,7 @@ class DeepEPDispatchMode(IntEnum):
class DeepEPBuffer: class DeepEPBuffer:
_buffer: Optional[Buffer] = None _buffer = None
_dispatch_mode: Optional[DeepEPDispatchMode] = None _dispatch_mode: Optional[DeepEPDispatchMode] = None
_hidden_size: Optional[int] = None _hidden_size: Optional[int] = None
_num_max_dispatch_tokens_per_rank: Optional[int] = None _num_max_dispatch_tokens_per_rank: Optional[int] = None
...@@ -165,7 +165,7 @@ class _DeepEPDispatcherImplBase: ...@@ -165,7 +165,7 @@ class _DeepEPDispatcherImplBase:
def combine_b(self, *args, **kwargs): def combine_b(self, *args, **kwargs):
raise NotImplementedError raise NotImplementedError
def _get_buffer(self) -> Buffer: def _get_buffer(self):
raise NotImplementedError raise NotImplementedError
......
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