Commit Graph
444 Commits
Author SHA1 Message Date
94371dbbd6 [feature] Ascend NPU graph support (#8027)
Co-authored-by: ronnie_zheng <zl19940307@163.com>
Co-authored-by: yezhifeng (D) <y00897525@china.huawei.com>
Co-authored-by: anon189Ty <Stari_Falcon@outlook.com>
Co-authored-by: Maksim <makcum888e@mail.ru>
Co-authored-by: ssshinigami <44640852+ssshinigami@users.noreply.github.com>
2025-08-16 17:25:17 -07:00
Trevor Morris eff4eb3fdd Add fp4 quantize before all-gather for Flashinfer cutlass MoE DP (max throughput) (#7667) 2025-08-15 22:08:11 -07:00
Cheng Wan 84b006b278 Cleanup MoE Refactor (#9223) 2025-08-15 02:28:33 -07:00
Cheng Wan 295895120d [6/N] MoE Refactor: Cleanup MoE-related configs (#8849) 2025-08-14 21:14:53 -07:00
Cheng Wan b87aacb5c5 [DP Attention] Refactor: adding some utility functions (#9136) 2025-08-13 21:08:06 -07:00
Faraz f508cd3cb7 TRTLLM-MLA FP8 path (#8638)
Signed-off-by: Faraz Khoubsirat <58580514+farazkh80@users.noreply.github.com>
2025-08-11 14:02:13 -07:00
Xiaoyu Zhang 44e86480e8 fuse allreduce and residual_rmsnorm (#8731) 2025-08-11 13:50:53 -07:00
Trevor Morris c0e84297c2 Use reduce scatter for DP (#8539) 2025-08-06 16:21:26 -07:00
d4bf5a8524 Support OCP MXFP4 quantization on AMD GPUs (#8255)
Co-authored-by: wunhuang <wunhuang@amd.com>
Co-authored-by: Hubert Lu <Hubert.Lu@amd.com>
2025-08-04 18:14:52 -07:00
azhurkevichandCheng Wan 915140fd18 [NVIDIA] Add Low Latency NVFP4 decode kernels from Flashinfer (#8552)
Co-authored-by: Cheng Wan <cwan@x.ai>
2025-08-04 03:10:02 -07:00
Nicolas Castet 82e6c3a65a Add support for NCCL symmetric memory for TP allreduces (#8238) 2025-08-01 23:30:55 +00:00
Cheng Wan 6c88f6c8d9 [5/N] MoE Refactor: Update MoE parallelism arguments (#8658) 2025-08-01 01:20:03 -07:00
Baizhou Zhang e7e5a3050a Update batch size limitation of dsv3_router_gemm kernel to 16 (#8051) 2025-08-01 11:53:31 +08:00
Kaixi Houandkushanam aa4c66b564 [NVIDIA] Enable Flashinfer MoE blockscale fp8 backend for TP MoE (#8450)
Co-authored-by: kushanam <42385577+kushanam@users.noreply.github.com>
2025-07-31 19:56:34 -07:00
Faraz 4b04998d38 TRTLLM Gen MLA Decode Kernel Integration (same as #7938) (#8632)
Signed-off-by: Faraz Khoubsirat <58580514+farazkh80@users.noreply.github.com>
2025-07-31 16:03:40 -07:00
Cheng Wan 9effeb5bdd Support EPLB in FusedMoE (#8448) 2025-07-29 16:02:41 -07:00
Cheng Wan 9c138a0445 [3/N] MoE Refactor: Simplify DeepEP Output (#8421) 2025-07-28 11:37:17 -07:00
2810338401 [feat] Support different attention backends for prefill and decode (#6338)
Co-authored-by: tianqilin.99 <tianqilin.99@bytedance.com>
Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>
2025-07-28 11:42:29 +08:00
Kaixi Hou 85486b6f6f [NVIDIA] Add Flashinfer MoE blockscale fp8 backend (#8036) 2025-07-27 00:34:41 -07:00
Cheng Wan c0fb25e949 DP Enhancement (#8280) 2025-07-24 21:36:21 -07:00
Ke Bao 6936be3221 Remve router gemm output dtype conversion (#8204) 2025-07-21 15:37:00 +08:00
Atream a589a07167 fix moe gate dtype, fix tbo, fix fake dispatch (#7825) 2025-07-19 22:13:46 -07:00
Cheng Wan 15ad6c9086 [1/N] MoE Refactor: refactor select_experts (#7966) 2025-07-19 00:51:15 -07:00
Hubert Lu 7750b91ca8 [AMD] Add triton awq_dequantize kernel to support AWQ on ROCm (#7661) 2025-07-18 14:27:25 -07:00
Hongbo XuandAniZpZ 1f76fc8747 [3/n] chore: decouple AWQ implementation from vLLM dependency (#8113)
Co-authored-by: AniZpZ <zhuangsen.zp@antgroup.com>
2025-07-18 11:45:22 -07:00
strgrbandZhang Kaihong 194841e329 remove kv_a.congigous in DeepseekV2AttentionMLA (#8058)
Co-authored-by: Zhang Kaihong <zhangkaihong.zkh@alibaba-inc.com>
2025-07-15 18:20:41 -07:00
Albert 38216cf049 concurrently load weights of DeepseekV2ForCausalLM (#7943)
Signed-off-by: Tianyu Zhou <albert.zty@antgroup.com>
2025-07-15 13:41:19 -07:00
Chunyuan WU 07452cbe8e [CPU] fix no attribute 'can_fuse_mlp_allreduce' error (#8010) 2025-07-14 01:32:43 -07:00
Cheng Wan 98aa836bbf Overlap the gating function with shared experts in DeepSeek (#7978) 2025-07-12 13:41:50 -07:00
fzyzcjy 2a2d3478af Fix wrong gemm branch cause 250us slower (#7969) 2025-07-11 19:45:09 -07:00
Xiaoyu Zhang aa2056091a delete uselese code caused by fuse allreduce+add_rmsnorm pr (#7970) 2025-07-11 19:43:38 -07:00
Xiaoyu Zhang 49a5915f53 [ready b200] fuse allreduce+add_rmsnorm in prepare_attention + mlp module (#7775) 2025-07-10 15:12:39 -07:00
Yineng Zhang b6b6268ccf Revert "Embedding parallel by attn_tp (#7623)" (#7880) 2025-07-08 22:03:09 -07:00
SijiaYang cb9d91ea8a feat: support DeepSeek-R1-W4AFP8 model with ep-moe mode (#7762)
Signed-off-by: yangsijia.614 <yangsijia.614@bytedance.com>
2025-07-07 14:47:21 -07:00
JieXin Liang 54411f6afa fix: disable dsv3_router_gemm in dsv3_nextn (#7793) 2025-07-05 19:01:01 -07:00
JieXin Liang c04a8a820b [fix] fix misusing of is_cuda (#7790) 2025-07-05 04:02:14 -07:00
Cheng Wan 8fc910db03 DP Attention with Auto DeepEP Dispatch (#7222) 2025-07-05 01:54:24 -07:00
Gang Chen ef8a29c429 Embedding parallel by attn_tp (#7623) 2025-07-04 23:21:56 -07:00
Cheng Wan cb432f1770 saving hidden_states.clone() (#7705) 2025-07-04 20:07:42 -07:00
Chunyuan WU 9fcc9a80e7 [CPU] refine CPU integration code (#7647) 2025-07-03 09:51:09 -07:00
JieXin Liang ac49dac009 [fix] fix dsv3_router_gemm filter (#7750) 2025-07-03 09:25:32 -07:00
1e0e549766 Ascend attention backend(PA&MLA) (#7722)
Co-authored-by: Maksim <makcum888e@mail.ru>
Co-authored-by: VDV1985 <vladdv85@mail.ru>
2025-07-03 09:23:19 -07:00
AniZpZ b58226510f fix dsv3 fused proj check (#7738) 2025-07-03 01:52:44 -07:00
AniZpZ 84f2e4a0f8 fix awq and dsv3 fused gemm compatible (#7735) 2025-07-02 22:56:57 -07:00
Baizhou Zhang 88f484ce4c Apply dsv3 router gemm kernel for deepseek-r1 fp4 (#7677) 2025-07-02 12:30:18 -07:00
Ke Bao 00aec6ad6c Apply dsv3_fused_a_gemm kernel (#7635) 2025-07-01 22:32:05 -07:00
fzyzcjy 0c9c6c75a8 Move files related to EPLB (#7580) 2025-06-29 15:39:38 -07:00
Chunyuan WUandZheng, Beilei <beilei.zheng@intel.com> a5317b2fd3 [CPU] add optimizations for INT8 and FP8 DeepSeek (#6769)
Co-authored-by: Zheng, Beilei <beilei.zheng@intel.com>
2025-06-27 19:04:29 -07:00
valarLip 802815e40b take aiter get_rope back (#7521) 2025-06-25 02:03:33 -07:00
Chunyuan WUandThien Tran 7eb47b0f3d [CPU] [BF16] Call fused_experts_cpu, weight_packed_linear and bmm_cpu kernel in DeepSeek model (#6641)
Co-authored-by: Thien Tran <gau.nernst@yahoo.com.sg>
2025-06-25 01:43:33 -07:00