29ecc2e602247f512c1d954c7b51f23b2d5c7166
Key difference from the reverted batched approach: - Does NOT use torch::mm in a C++ loop (that was the reverted commit) - Uses ixformer_torch_ext::ixformer_linear — the base image's optimized GEMM - Same kernel the competitor (sub 168) uses via corex_moe.py - Eliminates Python interpreter + dispatcher overhead per expert - Links against _ixformer_torch.cpython-310.so (already in base image) Decode: 1 Python call → 8 C++ ixformer_linear (vs 8 Python F.linear) Prefill: 1 Python call → 64 C++ ixformer_linear (vs 64 Python F.linear)
project_6
Description
Languages
C++
41.8%
Cuda
31.6%
Python
22.2%
C
2.1%
CMake
1.1%
Other
1.1%