flux
flux copied to clipboard
[QUESTION]flux/src/moe_ag_scatter/ths_op/gemm_grouped_v3_ag_scatter.cc你好,如果我需要在moe_ag的allgather之后对input_buffer做一个transpose(0,1),需要怎么修改这里的代码?需要改flux/src/moe_ag_scatter/sm90_gemm_array_warpspecialized_cooperative.hpp里面的代码吗?
Your question Ask a clear and concise question about Flux.