Commit Graph
42 Commits
Author SHA1 Message Date
Xiaoyu Zhang 50f28f65a0 fix typo in deep gemm benchmarking(#3991) 2025-03-02 00:34:00 -08:00
Xiaoyu Zhang 90a55e2566 add deepgemm and sglang fp8 block-wise gemm benchmark (#3893) 2025-03-01 23:01:58 -08:00
Xiaoyu Zhang 55a7ec388f use warp shuffle style reduce and flashinfer vectorize (#3628) 2025-02-19 20:53:51 +08:00
Xiaoyu Zhang c38f3aed24 support multi-gpu block-gemm tuning (#3639) 2025-02-18 00:00:35 +08:00
Xiaoyu Zhang 3efbdf68b9 fix sgl-kernel codestyle (#3563) 2025-02-14 18:05:52 +08:00
Xiaoyu Zhang f076328bb7 fix moe_align_kernel shm init not sync bug (#3534) 2025-02-13 16:47:00 +08:00
Xiaoyu Zhang 45e3a7bc41 use sgl_per_token_group_quant_fp8 kernel (#3493) 2025-02-12 18:40:42 +08:00
Xiaoyu Zhang 693c2600e0 refine deepseek_v3 launch server doc (#3522) 2025-02-12 17:27:07 +08:00
Xiaoyu Zhang bb418ced80 optimize per token group quant fp8 (#3490) 2025-02-11 22:19:05 +08:00
Xiaoyu Zhang 2f47d710ae refine some typo (#3473) 2025-02-10 23:35:44 +08:00
Xiaoyu Zhang cdae77b03d optimize moe_align_kernel cuda (#3347) 2025-02-07 00:53:46 +08:00
Xiaoyu Zhang ad3499858e clean moe align block kernel code and add acc test (#3332) 2025-02-06 16:42:36 +08:00
Xiaoyu Zhang 3c8ac78dc1 optimize test_fused_moe style (#3268) 2025-02-03 18:56:18 +08:00
Xiaoyu Zhang 81262c7b72 clean up useless file (#3192) 2025-01-28 14:29:30 +08:00
Xiaoyu Zhang 5d9d15e70f support fp32 in sampling_scaling_penalties kernel (#3121) 2025-01-25 16:52:17 +08:00
Xiaoyu Zhang e0cd65c2b6 [hotfix] fix test_sampling_scaling_penalties.py ci test (#3084) 2025-01-24 00:33:59 +08:00
Xiaoyu Zhang f1b6861828 use flashinfer vec_dtypes in sgl_kernel (#3083) 2025-01-23 22:19:04 +08:00
Xiaoyu Zhang ac2dc35d0e support lightning_attention_decode in sgl-kernel for MiniMax-Text-01 (#3030) 2025-01-23 15:29:20 +08:00
Xiaoyu Zhang 83452dbb4a fix file name spelling mistake and useless variable in minmax-text-01-lightning_attention (#2971) 2025-01-18 18:56:13 -08:00
Xiaoyu Zhang c2f212d672 optimize MiniMax-Text-01 lightning_attn_decode triton (#2966) 2025-01-18 23:41:01 +08:00
Xiaoyu Zhang 78e974b2a5 [kernel] MiniMax-Text-01 decode lightning_attn with triton (#2920) 2025-01-16 12:51:38 -08:00
Xiaoyu Zhang ab31793661 [kernel] MiniMax-Text-01 prefill lightning_attn with triton (#2911) 2025-01-16 14:18:29 +08:00
Xiaoyu Zhang f005758f2b introduce CUB in sgl-kernel (#2887) 2025-01-14 19:48:59 +08:00
Xiaoyu Zhang d08c77c434 Sampling penalties memory interface (#2870) 2025-01-13 23:09:00 +08:00
Xiaoyu Zhang e2b16c4716 add sampling_scaling_penalties kernel (#2846) 2025-01-12 19:38:17 -08:00
Xiaoyu Zhang f0e15dc6ab [HotFix] fix fp8 scale load failed in tp>1 (#2837) 2025-01-11 14:34:26 +08:00
Xiaoyu Zhang 380930a959 add benchmark_moe_align_blocks (#2767) 2025-01-07 14:20:50 +08:00
Xiaoyu Zhang ded9fcd09a improve moe_align_kernel for deepseek v3 (#2735) 2025-01-06 00:28:22 +08:00
Xiaoyu Zhang 286cad3ee3 h200 tuning fused_moe_triton config for Mixtral 8x7B/8x22B and Qwen2 57BA14B (#2689) 2024-12-31 23:17:36 +08:00
Xiaoyu Zhang 9254a33ad4 avoid fused_moe_triton padding circular import (#2624) 2024-12-28 14:01:35 +08:00
Xiaoyu Zhang 9a23c48456 h100 tuning fused_moe_triton for qwen2 moe (#2560) 2024-12-26 03:13:31 -08:00
Xiaoyu Zhang 7d672d277b [kernel optimize] benchmark write_req_to_token_pool_triton and optimize kernel (#2509) 2024-12-22 02:31:02 -08:00
Xiaoyu Zhang a0592c059f [Benchmark] add a benchmark for hf/vllm/sglang rmsnorm (#2486) 2024-12-15 13:52:08 +08:00
Xiaoyu Zhang 3844feb9bb Add a unittest for fused_moe (#2416) 2024-12-08 22:46:10 -08:00
Xiaoyu Zhang 34b364e073 optimize cuda graph max_bs_settings on low-end gpus (#2360) 2024-12-06 01:13:04 -08:00
262e370f78 [benchmark] Add fused_moe_triton benchmark and tuning tools (#2225)
Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
Co-authored-by: HAI <hixiao@gmail.com>
2024-11-29 13:36:45 -08:00
Xiaoyu Zhang dd44173dad [Fused moe] add tuning fused configs for qwen2 57b and mixtral 8x7b (#2167) 2024-11-25 10:37:50 +08:00
Xiaoyu Zhang 023d0a73df fix small typos in docs (#2047) 2024-11-15 11:09:10 -08:00
Xiaoyu Zhang eff468dd5a fix test_embedding_models prompt length too long's bug (#2015) 2024-11-12 23:21:16 +08:00
Xiaoyu Zhang a1bd719031 fix a bug in v1_embeeding_request (#2014) 2024-11-12 16:49:45 +08:00
Xiaoyu Zhang 027e65248f support echo=true and logprobs in openai api when logprobs=1 in lm-evaluation-harness (#1998) 2024-11-11 23:21:20 -08:00
Xiaoyu Zhang 605972195b check user-specified model_max_len with hf derived max_model_len (#1778) 2024-10-24 12:40:36 -07:00