3 people e8c24a7695 [Kernel] Vectorized fp32 moe_sum reduction and support any topk (#46643) ...

Signed-off-by: mgoin <mgoin64@gmail.com>
Co-authored-by: Claude <noreply@anthropic.com>

2026-06-25 14:02:28 -07:00

..

marlin_moe_wna16

[11a/n] Migrate Marlin kernels to torch stable ABI (#45176)

2026-06-11 21:22:37 -07:00

permute_unpermute_kernels

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00

dsv3_router_gemm_bf16_out.cu

[Kernel] GLM5 Router GEMM (#46385)

2026-06-23 22:54:50 -07:00

dsv3_router_gemm_entry.cu

[Kernel] GLM5 Router GEMM (#46385)

2026-06-23 22:54:50 -07:00

dsv3_router_gemm_float_out.cu

[Kernel] GLM5 Router GEMM (#46385)

2026-06-23 22:54:50 -07:00

grouped_topk_kernels.cu

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00

moe_align_sum_kernels.cu

[Kernel] Vectorized fp32 moe_sum reduction and support any topk (#46643)

2026-06-25 14:02:28 -07:00

moe_ops.h

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00

moe_permute_unpermute_op.cu

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00

moe_wna16_utils.h

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00

moe_wna16.cu

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00

moeTopKFuncs.cuh

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00

topk_softmax_kernels.cu

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00

topk_softplus_sqrt_kernels.cu

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00

torch_bindings.cpp

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00