71d39a1c7e517ff7f2334e00b012f6cbf7b9045e
Two fused kernels to replace Python loops in MoE prefill path:
1. moe_compute_index: histogram + CUB BlockScan prefix_sum + place
replaces: argsort + bincount + CPU sync
2. moe_combine_result: fused weighted sum of expert outputs
replaces: view + multiply + sum
Source: xllm_latest/core/kernels/cuda/moe/{moe_compute_index.cu, moe_combine.cu}
Adapted: removed xllm framework deps, added pybind11 wrapper
Verify on real BI-V100: python3 verify_moe_index_combine.py
project_6
Description
Languages
C++
41.8%
Cuda
31.6%
Python
22.2%
C
2.1%
CMake
1.1%
Other
1.1%