Commit Graph
16 Commits
Author SHA1 Message Date
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