About the version of DeepEP for mega_moe_kernel

[NiuMa-1234]

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!

[engineer1109]

+1

[thisjiang]

+1

[zheanxu]

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.