Commit Graph
100 Commits
Author SHA1 Message Date
Xiaoyu Zhangandgithub-actions[bot] 407cb3ce1e [CI tiny fix] Enhance robustness of vision chunked prefill test with ROUGE-L metric (#13793)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
2025-11-25 15:41:14 +08:00
Xiaoyu Zhang ecefc7904f [sgl-kernel Code Clean] Remove useless lightning_attention kernel (#13819) 2025-11-24 18:26:25 +08:00
Xiaoyu Zhang 9ea1953331 [Doc] Refine fused_moe_triton configs doc (#13820) 2025-11-23 19:09:41 -08:00
Xiaoyu Zhang b964ce61d6 [DeepEP] Add SGLANG_DEEPEP_BF16_DISPATCH env var in Normal mode (#13787) 2025-11-23 17:32:43 +08:00
Xiaoyu Zhang a34d3abb54 [Clean code] Compressed_tensors_moe code clean (#13719) 2025-11-21 18:15:46 +08:00
Xiaoyu Zhang bfcf15a129 [opt kimi k2 4 / n] Delete useless pad kernel in sgl_moe_align_block_size (#13587) 2025-11-21 13:16:42 +08:00
fb04d43428 [kimi k2 thinking] Avoid useless torch.zeros_ (#13596)
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
2025-11-21 13:15:27 +08:00
Xiaoyu Zhang dc69462456 [CI fix] Fix image download failures in VLM CI tests (#13613) 2025-11-20 11:18:06 +08:00
Xiaoyu Zhang 820e13c9c1 [opt kimi k2 3/n] opt kimi_k2 moe_fused_gate kernel (#13374) 2025-11-18 15:36:31 +08:00
Xiaoyu Zhang 7b44526038 [Ci tiny fix] Lower score threshold in evaluation test (#13443) 2025-11-18 00:45:20 +08:00
Xiaoyu Zhang 8b5e2c5368 [Tiny fix] Fix bench_speculative.py run bug (#13416) 2025-11-17 18:58:19 +08:00
Xiaoyu ZhangandYuan Luo 95f43669b5 [2 / 2] apply sgl-kernel weak_ref_tensor (#12978)
Co-authored-by: Yuan Luo <yuan.luo@hotmail.com>
2025-11-16 21:24:35 +08:00
Xiaoyu Zhang 50691d7b49 [opt kimi k2 2/n] apply kimi k2 thinking moe_fused_gate (#13332) 2025-11-16 21:20:08 +08:00
Xiaoyu Zhang 1d3d42bda0 [opt kimi k2 1 / n] Add kimi k2 moe fused gate (#13287) 2025-11-15 17:14:19 +08:00
Xiaoyu Zhang a1cb717d0b Opt kimi_k2_thinking biased topk module (#13150) 2025-11-12 16:52:06 -08:00
Xiaoyu Zhang fe92d4d88e [CI] Auto format code (#13053) 2025-11-10 23:30:29 -08:00
Xiaoyu Zhang 9caca6a45c [PieceWise CUDA Graph] Support awq/gptq model in piecewise cudagraph (#12518) 2025-11-11 11:56:15 +08:00
Xiaoyu Zhang f18ec927f3 fix tuning_fused_moe_triton_sep tool per_channel_quant bug (#13027) 2025-11-11 10:33:54 +08:00
Xiaoyu Zhang 547de8c774 [1 / 2] register weak_ref_tensor in sgl-kernel (#12999) 2025-11-10 22:12:59 +08:00
Xiaoyu Zhang 05559a4a90 Support hidden_dim % 4 == 0 in per_token_quant_fp8 (#12883) 2025-11-10 17:13:14 +08:00
fc84b0730c [Refactor] Refactor fused_moe_triton tuning tools: extract shared utils, add EP/MLLM support, reduce overhead (#12440)
Co-authored-by: xu-yfei <xu-yfei@users.noreply.github.com>
Co-authored-by: Yongfei Xu <xuyongfei.xyf@antgroup.com>
2025-11-06 20:54:42 +08:00
Xiaoyu Zhang 95191ebdca Migrate weak_ref_tensor to sgl-kernel (#12505) 2025-11-02 10:55:39 +08:00
Xiaoyu Zhang d8fcbaa38d [CI Monitor] Fix ci_monitor perf analyzer bug (#12281) 2025-10-30 09:47:12 -07:00
Xiaoyu Zhang 04e5b6faa7 Revert "Triton fused_moe_kernel support ep moe tuning" (#12377) 2025-10-30 07:12:06 -07:00
Xiaoyu Zhang 52694b60da Triton fused_moe_kernel support ep moe tuning (#12343) 2025-10-29 23:16:09 +08:00
Xiaoyu Zhang 334543ff3b Add continuous_usage_stats support for streaming responses (#12241) 2025-10-29 10:01:23 +08:00
Xiaoyu Zhang d0cff78f54 [CI] Add ci monitor balance workflow (#11962) 2025-10-25 12:14:36 -07:00
Xiaoyu Zhang 8470133852 [b200] fix piecewise cuda graph launch bug (#12067) 2025-10-24 22:36:39 +08:00
Xiaoyu Zhang 8374a96e49 piecewise cuda graph support qwen3-moe (#11845) 2025-10-21 10:55:49 +08:00
Xiaoyu Zhang 984fbeb16b Revert "[CI Monitor] Ci monitor only deal with main branch in default" (#11846) 2025-10-19 22:06:40 -07:00
Xiaoyu Zhang 24ed3f32c0 fix(ci): Fix CI Monitor limit parameter and add CI Analysis to summary (#11832) 2025-10-19 18:08:34 -07:00
Xiaoyu Zhang 88a6f9dab5 bench_serving support PD Disaggregation (#11542) 2025-10-13 19:43:26 -07:00
Xiaoyu Zhang 8e51049f56 [CI Monitor] Ci monitor only deal with main branch in default (#11538) 2025-10-13 13:50:04 -07:00
Xiaoyu Zhang 6806c4e63e [CI monitor] Improve CI analyzer: fix job failure tracking and add CUDA-focused filtering (#11505) 2025-10-13 13:31:09 +08:00
Xiaoyu Zhang 6f16bf9d9d [Ci Monitor] Auto uploaded performance data to sglang_ci_data repo (#10976) 2025-09-29 16:17:27 +08:00
Xiaoyu Zhang 11965b0daf Fix sgl-kernel benchmark dead code (#11022) 2025-09-29 15:06:40 +08:00
Xiaoyu Zhang 2387c22b56 Ci monitor support performance (#10965) 2025-09-27 09:11:21 +08:00
05a3526654 Restruct gpu_memory_settings in a unify function and relax max_cuda_graph_bs (#10372)
Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
Co-authored-by: sglang-bot <sglangbot@gmail.com>
2025-09-26 15:10:49 -07:00
Xiaoyu Zhang c4197e99bb [ci] add ci-monitor workflow (#10898) 2025-09-25 19:29:47 -07:00
Xiaoyu Zhang c1f39013b7 [ci feature] add ci monitor (#10872) 2025-09-24 23:16:29 -07:00
Xiaoyu Zhang c4e314f986 Restruct sgl-kernel benchmark (#10861) 2025-09-25 07:45:25 +08:00
Xiaoyu Zhang 37367da639 [fix CI] Fix logical condition in fused MoE layer for compressed tensor quantization (#10299) 2025-09-10 23:54:09 -07:00
Xiaoyu Zhang b1fb7e458c [benchmark] add flashinfer_allreduce_fusion benchmark (#9937) 2025-09-03 16:31:01 +08:00
Xiaoyu Zhang a1e5d78115 fix parallel_state.py current_platform bug (#9919) 2025-09-02 03:17:15 -07:00
Xiaoyu Zhang b5245064f6 [code style] restruct fused_moe to avoid very long single file (#9878) 2025-09-02 11:04:27 +08:00
Xiaoyu Zhang f96413c444 Refactor allreduce add rmsnorm pattern (#9278) 2025-08-20 02:03:08 -07:00
Xiaoyu Zhang 63d82a776a refine mxfp4 shuffling log (#9194) 2025-08-14 10:57:29 -07:00
Xiaoyu Zhang 44e86480e8 fuse allreduce and residual_rmsnorm (#8731) 2025-08-11 13:50:53 -07:00
Xiaoyu Zhang a886564a18 fix flashinfer allreduce fusion import bug (#9007) 2025-08-09 13:47:05 -07:00
Xiaoyu Zhang 0d1e27a0c5 Better optimization log for gpt-oss model (#8953) 2025-08-08 00:11:48 -07:00
Xiaoyu Zhang 76915d68a8 Fix enable flashinfer mxfp4 moe bf16 check (#8950) 2025-08-07 22:52:09 -07:00
Xiaoyu Zhang 3ae33fcd0a Fix hopper launch gpt-oss model illegal memory (#8908) 2025-08-07 10:02:40 -07:00
Xiaoyu Zhang 47824c1488 [Perf] Auto enable best flashinfer mxfp4 kernel in b200 (#8898) 2025-08-07 01:08:41 -07:00
Xiaoyu Zhang 4373df5525 add flashinfer mxfp4 (#8847) 2025-08-06 16:23:41 -07:00
Xiaoyu Zhang f57d2dc162 [sgl-kernel] avoid per_token_quant_fp8.cu hardcode sm_count (#8738) 2025-08-04 12:55:57 +08:00
Xiaoyu ZhangandKe Bao 7a4309cc8a [sgl-kernel performace] fix fp8 quant kernels dispatch __nv_fp8_e4m3 bug to improve performance 10%-20% (#8499)
Co-authored-by: Ke Bao <ispobaoke@gmail.com>
2025-07-29 23:31:54 +08:00
Xiaoyu Zhang 2262369905 Revert "[kernel] opt moe align block kernel by block/warp scan algorithm" (#8457) 2025-07-28 01:35:43 -07:00
Xiaoyu Zhang 9045cc1eb8 [torch.compile bug] avoid biased_grouped_topk_impl func repeatedly triggering torch.compile in forward pass (#8353) 2025-07-25 21:17:47 +08:00
Xiaoyu Zhang a167fd0bcb [code style] Clean dead triton kernel code in fused_moe and useless vllm_ops import (#8310) 2025-07-24 14:38:30 +08: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
Xiaoyu Zhang 2e7ab862e3 Fix illegal memory in trtllm allreduce fusion (#7864) 2025-07-08 11:47:17 -07:00
Xiaoyu Zhang 8e64140e35 [b200] support trt-llm allreduce fuse rms_norm_add kernel (#7621) 2025-07-02 19:36:20 -07:00
Xiaoyu Zhang ff2e9c9479 Add small requirements for benchmark/parse_result tools (#7671) 2025-06-30 21:52:20 -07:00
Xiaoyu Zhang 8ecad0b16f [benchmark] fbgemm benchmark support bandwidth report and support fbgemm_cutlass_gmm (#7422) 2025-06-24 09:44:55 -07:00
Xiaoyu Zhang 0ae1e9a755 refine fused_moe benchmark (#7221) 2025-06-15 21:21:32 -07:00
Xiaoyu Zhang 3712abfaf9 Fuse routed scaling factor in deepseek (#6970) 2025-06-08 15:24:24 -07:00
Xiaoyu Zhang fa3592cfeb rebase h20 fused_moe config (#6966) 2025-06-08 05:01:34 -07:00
Xiaoyu Zhang 515ef4facb Fuse routed scaling factor in topk_reduce kernel (#6220) 2025-06-07 11:06:50 -07:00
Xiaoyu Zhang bae4fdc7ab add fbgemm moe grouped gemm kernel benchmark (#6924) 2025-06-07 02:57:30 -07:00
Xiaoyu Zhang 8b5f83ed3b reduce torch.zeros overhead in moe align block size kernel (#6369) 2025-06-07 02:47:36 -07:00
Xiaoyu Zhang 2a413829f4 Add triton version as a fused_moe_triton config search key to avoid performace decrease in different Triton version (#5955) 2025-06-07 02:43:50 -07:00
Xiaoyu ZhangandJieXin Liang bd75690f4e fix ep_moe_reorder kernel bugs (#6858)
Co-authored-by: JieXin Liang <Alcanderian@users.noreply.github.com>
2025-06-04 19:13:59 +08:00
Xiaoyu Zhang 076103535c fix log_info_on_rank0 error when run benchmark (#6260) 2025-05-28 00:20:01 -07:00
Xiaoyu Zhang 9f2c9568f0 [doc] add a note for --n-share-experts-fusion args (#6154) 2025-05-11 23:18:38 -07:00
Xiaoyu Zhang d25398cbc8 fix custom_allreduce namespace (#6039) 2025-05-06 19:13:06 -07:00
Xiaoyu Zhang 5bb0accbcf cutlass 3.9 supported to improve fp8_blockwise_gemm (#5820) 2025-04-28 21:52:36 -07:00
Xiaoyu Zhang 1cc326032d simplify fused_moe config logging (#5801) 2025-04-28 17:04:54 -07:00
Xiaoyu Zhang e132cba2a8 fused moe triton tuning script support qwen3 (#5842) 2025-04-28 09:13:04 -07:00
Xiaoyu Zhang ef15dcda26 Add a doc to fix sgl-kernel build link error in py39 with ccache (#5809) 2025-04-27 21:34:27 -07:00
Xiaoyu Zhang ad4df30741 Dockerfile.dev pip scikit_build_core (#5807) 2025-04-27 21:14:20 -07:00
Xiaoyu Zhang 18ce468d56 update triton 3.2.0 h200 fused moe triton config and add warning about triton fused_moe_kernel performance degradation due to different Triton versions. (#5740) 2025-04-25 16:24:59 -07:00
Xiaoyu Zhang d9dd529854 enable DeepSeek V3 shared_experts_fusion in sm90 (#5571) 2025-04-20 12:46:42 -07:00
Xiaoyu Zhang d58e354472 simplify the control logic for using shared experts fusion (#5504) 2025-04-19 13:17:35 -07:00
Xiaoyu Zhang bf86c5e990 restruct compressed_tensors_w8a8_fp8 (#5475) 2025-04-19 04:52:15 -07:00
Xiaoyu Zhang bed05878f6 fix kimi vl running bug after rebase main (#5461) 2025-04-18 00:17:34 -07:00
Xiaoyu Zhang 8e09b37077 Sgl kernel fused_moe_gate support n_shared_experts (#5440) 2025-04-17 23:05:15 -07:00
Xiaoyu Zhang 06a1656e02 [doc] Update benchmark_and_profiling.md (#5449) 2025-04-15 23:27:34 -07:00
Xiaoyu ZhangandYineng Zhang 38076dea84 apply fused moe gate in ds v3/r1 (#5371)
Co-authored-by: Yineng Zhang <me@zhyncs.com>
2025-04-14 16:24:26 -07:00
Xiaoyu Zhang 3e4794aad8 refine fused_moe tuning docs (#5294) 2025-04-12 10:01:13 -07:00
Xiaoyu Zhang 690ec20587 Delete python/sglang/srt/layers/moe/fused_moe_triton/configs/E=257,N=… (#5321) 2025-04-12 10:00:03 -07:00
Xiaoyu Zhang 60bcbf2a35 remove moe_align_block_size torch.zeros in small batch/expert mode (#5298) 2025-04-11 12:13:55 -07:00
Xiaoyu Zhang f730362ee2 reduce moe_align_block_size_kernel small batch mode overhead (#5086) 2025-04-09 17:59:35 -07:00
Xiaoyu Zhang 87eddedfa2 [ci] fix ci test fused_moe op (#5102) 2025-04-09 08:52:46 -07:00
Xiaoyu Zhang db452760e5 [ci] fix llama4 ci error (#5126) 2025-04-07 21:15:46 +08:00
Xiaoyu Zhang 924ca7c92c Add DeepSeek V3/R1 shared experts fusion (#4918) 2025-04-04 01:59:29 -07:00
Xiaoyu Zhang 772d2a191d try to fix ci oserror (#5024) 2025-04-03 02:45:05 -07:00
Xiaoyu Zhang 2c8fd99363 [sgl-kernel] per token group quant support COLUMN MAJOR (#4817) 2025-04-02 18:29:59 -07:00
Xiaoyu Zhang e9c6ce461d sgl scaled_fp8_quant support output padding (#4861) 2025-04-02 23:53:57 +08:00
Xiaoyu Zhang 04e3ff6975 Support compressed tensors fp8w8a8 (#4743) 2025-03-26 13:21:25 -07:00