flux icon indicating copy to clipboard operation
flux copied to clipboard
verified publisher icon bytedance

[QUESTION] /src/moe_ag_scatter/test/test_moe_ag_scatter.cc only shows a V3 version

Open hMingXuuu opened this issue 7 months ago • 1 comments

Hey guys, /src/moe_ag_scatter/test/test_moe_ag_scatter.cc supports a v3 version of GemmGroupedAgScatterArguments. Can you provide an implementation of GemmGroupedV2AGScatterArguments? It will be convenient for testing on sm80 machines. Thank you.

hMingXuuu avatar May 04 '25 09:05 hMingXuuu

V2 and V3 are nearly the same. refer to https://github.com/bytedance/flux/blob/main/python/flux/cpp_mod.pyi#L649 here

houqi avatar May 04 '25 22:05 houqi