dylan
e2fc3f270f
fix: corex_batched_gemm use TCU OpClassTensorOp + Cu10 + float accum
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.
2026-08-15 12:35:56 +00:00
..
2026-08-15 05:22:31 +00:00
2026-08-14 07:48:52 +00:00
2026-08-10 09:12:05 +00:00
2026-08-15 06:15:17 +00:00
2026-08-15 12:35:56 +00:00
2026-08-13 03:19:39 +00:00
2026-08-13 03:19:39 +00:00
2026-08-15 07:00:09 +00:00
2026-08-12 01:33:24 +00:00
2026-08-10 04:01:21 +00:00
2026-08-15 06:15:17 +00:00
2026-08-12 01:33:24 +00:00
2026-08-10 07:50:28 +00:00