Xin Yang 8a3cd90af5

[Kernel] Add fused grouped_topk kernel for MoE (#23274) ...

Signed-off-by: Xin Yang <xyangx@amazon.com>
Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>

2025-08-25 11:47:52 -07:00

..

marlin_moe_wna16

[Kernel] [Quantization] Add MXFP4 and bias support for marlin kernel (#22428)

2025-08-14 11:23:22 -07:00

permute_unpermute_kernels

Fix CUDA permute/unpermute for use with DeepGemm Moe (#17934)

2025-07-27 07:08:00 -07:00

grouped_topk_kernels.cu

[Kernel] Add fused grouped_topk kernel for MoE (#23274)

2025-08-25 11:47:52 -07:00

moe_align_sum_kernels.cu

[perf] Speed up align sum kernels (#21079)

2025-07-21 11:19:23 -07:00

moe_ops.h

[Kernel] Add fused grouped_topk kernel for MoE (#23274)

2025-08-25 11:47:52 -07:00

moe_permute_unpermute_op.cu

[Kernel] CUTLASS MoE FP8: Integrate cuda moe permute/unpermute (#23045)

2025-08-20 10:35:26 -04:00

moe_wna16_utils.h

pre-commit autoupdate (#17380)

2025-04-29 06:46:55 -07:00

moe_wna16.cu

[BugFix] Accuracy fix for llama4 int4 - improperly casted scales (#16801)

2025-04-17 22:13:29 -07:00

topk_softmax_kernels.cu

[ROCm][Bugfix] Fix compilation error in topk softmax fused kernel (#22819)

2025-08-13 13:45:03 -07:00

torch_bindings.cpp

[Kernel] Add fused grouped_topk kernel for MoE (#23274)

2025-08-25 11:47:52 -07:00