Xin Yang 63227accf5 [Kernel] Add topk_sigmoid kernel (#31246) ...

Signed-off-by: Xin Yang <xyangx@amazon.com>

2026-01-21 22:49:51 +00:00

..

marlin_moe_wna16

Add thread_n=64 support to Marlin MoE (#32360)

2026-01-15 16:45:44 -08:00

permute_unpermute_kernels

[Perf] Only clone when needed for moe_permute (#32273)

2026-01-20 11:34:39 -05:00

dynamic_4bit_int_moe_cpu.cpp

[CPU]Parallelize over tokens in int4 moe (#29600)

2025-12-02 06:21:39 +00:00

grouped_topk_kernels.cu

[Perf] Optimize grouped topk kernel, 1.2%~2% E2E Throughput improvement (#32058)

2026-01-13 10:58:18 -08:00

moe_align_sum_kernels.cu

Lora MoE Align Improvements (#29257)

2025-12-09 10:35:16 +08:00

moe_ops.h

[Kernel] Add topk_sigmoid kernel (#31246)

2026-01-21 22:49:51 +00:00

moe_permute_unpermute_op.cu

[Perf] Only clone when needed for moe_permute (#32273)

2026-01-20 11:34:39 -05: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

[Kernel] Add topk_sigmoid kernel (#31246)

2026-01-21 22:49:51 +00:00

torch_bindings.cpp

[Kernel] Add topk_sigmoid kernel (#31246)

2026-01-21 22:49:51 +00:00