Yi Pan 45fdf1f7f3 Fix shared memory OOM on sm86 GPUs. (#4797)

2025-03-26 10:41:53 -07:00

..

allreduce

fix custom allreduce performance/accuracy problem (#4477)

2025-03-16 12:16:30 -07:00

attention

…

cutlass_extensions

…

elementwise

Optimize rope in sgl kernel (#4267)

2025-03-10 10:07:45 -07:00

gemm

Fix shared memory OOM on sm86 GPUs. (#4797)

2025-03-26 10:41:53 -07:00

moe

Add moe topk softmax templated from vllm (#4302)

2025-03-14 12:03:33 -07:00

speculative

[ROCm] Enable MTP (NextN) on AMD GPU (#4631)

2025-03-23 22:58:05 -07:00

torch_extension_rocm.cc

[ROCm] Enable MTP (NextN) on AMD GPU (#4631)

2025-03-23 22:58:05 -07:00

torch_extension.cc

Support FP4 gemm (1/2) (#3899)

2025-03-24 19:50:23 -07:00