e2fc3f270f91e355d3f8f7d02d7d9a9341564faa
Previous version used default SIMT path (25ms). Fixed version matches moe_cutlass_batched.cu (2.462ms): - ElementAccumulator = float (was half_t) - OpClassTensorOp (was default OpClassSimt) - arch::Cu10 (was default Sm61) - RowMajor layout (was ColumnMajor) - torch::sigmoid(x)*x instead of torch::silu (not in corex torch) Also fixed bind.cpp: removed col-major transposition logic, kernel now RowMajor so A(M,K) @ B(K,N) = C(M,N) directly.
project_6
Description
Languages
C++
41.8%
Cuda
31.6%
Python
22.2%
C
2.1%
CMake
1.1%
Other
1.1%