[GH-ISSUE #306] About the version of DeepEP for mega_moe_kernel #155

Closed
opened 2026-10-26 21:13:49 +00:00 by bsakaguchi · 3 comments
Owner

Originally created by @NiuMa-1234 on GitHub (Apr 17, 2026).
Original GitHub issue: https://github.com/deepseek-ai/DeepGEMM/issues/306

Thank you for your marvelous work! i've been trying to test the mega_moe kernel , and encountered an error about deepep:

ep_buffer = deep_ep.ElasticBuffer(
                ^^^^^^^^^^^^^^^^^^^^^
AttributeError: module 'deep_ep' has no attribute 'ElasticBuffer'

And I found that there's no ElasticBuffer in the latest DeepEP... Could you please give the version of DeepEP for the kernel? Many many thanks!

Originally created by @NiuMa-1234 on GitHub (Apr 17, 2026). Original GitHub issue: https://github.com/deepseek-ai/DeepGEMM/issues/306 Thank you for your marvelous work! i've been trying to test the mega_moe kernel , and encountered an error about deepep: ``` ep_buffer = deep_ep.ElasticBuffer( ^^^^^^^^^^^^^^^^^^^^^ AttributeError: module 'deep_ep' has no attribute 'ElasticBuffer' ``` And I found that there's no ElasticBuffer in the latest DeepEP... Could you please give the version of DeepEP for the kernel? Many many thanks!
Author
Owner

@engineer1109 commented on GitHub (Apr 17, 2026):

+1

<!-- gh-comment-id:4266449625 --> @engineer1109 commented on GitHub (Apr 17, 2026): +1
Author
Owner

@thisjiang commented on GitHub (Apr 24, 2026):

+1

<!-- gh-comment-id:4311501475 --> @thisjiang commented on GitHub (Apr 24, 2026): +1
Author
Owner

@zheanxu commented on GitHub (Apr 30, 2026):

Thank you for your kind words and for testing the mega_moe kernel!

The ElasticBuffer class is indeed a new feature introduced in DeepEP v2. The latest DeepEP release (v2) includes this API, but the older version does not. Please make sure you install the correct version — you can find it at the link you mentioned: DeepEP v2. After updating, the error should be resolved.

<!-- gh-comment-id:4350704701 --> @zheanxu commented on GitHub (Apr 30, 2026): Thank you for your kind words and for testing the mega_moe kernel! The `ElasticBuffer` class is indeed a new feature introduced in **DeepEP v2**. The latest DeepEP release (v2) includes this API, but the older version does not. Please make sure you install the correct version — you can find it at the link you mentioned: [DeepEP v2](https://github.com/deepseek-ai/DeepEP/pull/605). After updating, the error should be resolved.
Sign in to join this conversation.
No labels
pull-request
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
deepseek-ai/DeepGEMM#155
No description provided.