Commit Graph
249 Commits
Author SHA1 Message Date
Lianmin Zheng 2b0fc5941d [Minor] Code style improvements (#2355) 2024-12-04 19:02:08 -08:00
Jerry Zhang 9cc733b38c move apply_torchao_config_ to model_runner (#2342) 2024-12-04 17:26:42 -08:00
Ke Bao ec52464dde MLA prefill w/o weight absorption (#2349) 2024-12-05 01:50:28 +08:00
HAI b2986d7aa5 Adding SGLang FP8 Utils (#2348) 2024-12-04 03:01:33 -08:00
Lianmin Zhengandroot 1228f7ca69 Fix gptq for moe layers (#2300)
Co-authored-by: root <me@zhyncs.com>
2024-12-03 23:12:33 +08:00
Lianmin Zheng 07ec07ad1f Improve torch compile for fused moe (#2327) 2024-12-03 01:58:25 -08:00
Yineng ZhangandHandH1998 85e1a6f3aa Update model_loader deps and qqq quantization deps (#2220) (#2318)
Co-authored-by: HandH1998 <1335248067@qq.com>
2024-12-02 23:22:13 +08:00
Qun Yang 62c516ac45 Add a simple torch native attention backend (#2241) 2024-12-01 03:01:25 -08:00
Yineng Zhang 118b6af35e feat: add should_use_tensor_core (#2179) 2024-12-01 18:01:16 +08:00
Lianmin Zheng 4936be8acc Revert "Revert "[FEAT] Support GGUF format"" (#2287) 2024-11-30 22:14:48 -08:00
Lianmin Zheng 7e4c6dd8da Revert "[FEAT] Support GGUF format" (#2285) 2024-11-30 19:03:26 -08:00
Yang ZhengandYang Zheng 883c955489 [FEAT] Support GGUF format (#2215)
Co-authored-by: Yang Zheng(SW)(Alex) <you@example.com>
2024-11-30 00:44:48 -08:00
HAI cd51758fad Rename tuned MI300X config files for fused_moe_triton (#2228) 2024-11-27 21:18:51 -08:00
Lianmin Zheng dd5eba4c88 Remove fused_moe_grok (#2223) 2024-11-27 14:28:55 -08:00
Lianmin Zheng fb6e04a0c2 Use an env var SGLANG_SET_CPU_AFFINITY to set cpu affinity; turn it off by default (#2222) 2024-11-27 02:52:46 -08:00
Lianmin Zheng 6997e28f6e Revert "Use an env var SGLANG_SET_CPU_AFFINITY to set cpu affinity; turn it off by default" (#2221) 2024-11-27 02:02:01 -08:00
Lianmin Zheng a0e58740a8 Use an env var SGLANG_SET_CPU_AFFINITY to set cpu affinity; turn it off by default (#2217) 2024-11-27 01:13:41 -08:00
Lianmin Zheng ba4ee37fa4 Update sampler.py to skip the success check (#2197) 2024-11-26 00:58:57 -08:00
Yineng Zhang 55842eb81a feat: fused_moe fp8 monkey patch (#2174) 2024-11-25 17:06:36 +08:00
Lianmin Zheng 8e1adb8441 Allow overwrite flashinfer use_tensorcore (#2169) 2024-11-24 20:58:17 -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
Lianmin Zheng be0124bda0 Rename triton_fused_moe -> fused_moe_triton (#2163) 2024-11-24 08:12:35 -08:00
Yineng Zhang fa27161380 fix: use torch.sum for compatible (#2161) 2024-11-24 22:37:04 +08:00
Lianmin Zheng 5652c56535 Update CI threshold & Improve code style (#2159) 2024-11-24 06:29:38 -08:00
Yineng Zhang e3938b2f9c feat: update other MoE models deps (#2156) 2024-11-24 21:36:34 +08:00
Yineng Zhang d90c3d6b8b fix: resolve end-of-file-fixer (#2157) 2024-11-24 20:38:26 +08:00
Yineng Zhang 9e8f8fbf95 feat: update gitignore and add tuning config for FusedMoE (#2155) 2024-11-24 20:24:58 +08:00
Yineng Zhang b509db5832 feat: remove the dependency on FusedMoE (#2153) 2024-11-24 20:09:27 +08:00
Lianmin Zheng a78d8f8db3 [CI] Fix test cases (#2137) 2024-11-23 01:00:07 -08:00
Ke Bao c5f865013e Fix grid size in Triton decoding kernel (#2134) 2024-11-23 16:51:46 +08:00
Xuehai Pan 62a4a339eb docs: fix module docstrings and copyright headers (#2077) 2024-11-22 22:16:53 +08:00
HAI f35cb46cc3 ROCm: Fix MoE padding for none FP8 cases (#2111) 2024-11-21 12:23:21 -08:00
Jerry Zhang 5c6a41facf Error out when torchao-config option is not recognized (#2107) 2024-11-20 17:37:28 -08:00
Lianmin Zheng 722530fa01 Enable overlap scheduler by default for the triton attention backend (#2105) 2024-11-20 02:58:35 -08:00
Lianmin Zheng ffd20fcd03 Make constrained decoding work for overlap scheduler (#2095) 2024-11-19 15:04:43 -08:00
Yineng Zhang 766192610e feat: update torch 2.5.1 (#2069) 2024-11-18 21:29:13 +08:00
4af3f889fc Simplify flashinfer indices update for prefill (#2074)
Co-authored-by: kavioyu <kavioyu@tencent.com>
Co-authored-by: kavioyu <kavioyu@gmail.com>
2024-11-18 00:02:36 -08:00
Lianmin Zheng df7fe4521a Crash the CI jobs on model import errors (#2072) 2024-11-17 22:18:11 -08:00
Lianmin Zheng ebaa2f3199 Rename arguments --disable-nan-detection to --enable-nan-detection (#2066) 2024-11-17 16:53:44 -08:00
Lianmin Zheng c1f401fc58 Revert "chore: update torch v2.5.1" (#2063) 2024-11-17 15:29:38 -08:00
Yineng Zhang 3b878863f7 chore: update torch v2.5.1 (#1849) 2024-11-18 00:06:00 +08:00
Lianmin ZhengandHaotian Liu f719d9aebc Launch dp ranks in parallel (#2053)
Co-authored-by: Haotian Liu <6631389+haotian-liu@users.noreply.github.com>
2024-11-16 17:39:39 -08:00
Lianmin Zheng edad373135 Fix illegal memory access in overlap mode & Use more fused triton kernels for building meta data (#2051) 2024-11-16 16:14:23 -08:00
Ke Bao 976bc302e5 Support DP MLA (#1970) 2024-11-16 09:01:43 +00:00
Lianmin Zheng c3eac1b010 Fix torch.compile for MoE (#2033) 2024-11-14 01:30:24 -08:00
HAI 087ab83223 [Performance, Triton] Optimize over mask compute to tl.load in fused_moe_kernel (#1980) 2024-11-10 18:54:43 -08:00
HAI 67c424cce3 [Performance, Triton Kernel Args] extend_attention, optimize kern args to _fwd_kernel (#1941) 2024-11-07 18:24:02 -08:00
Chayenne c77c1e05ba fix black in pre-commit (#1940) 2024-11-08 07:42:47 +08:00
Xuehai Pan a5e0defb5a minor: Add basic editorconfig and pre-commit hooks to enforce style for whitespaces (#1926) 2024-11-06 13:46:04 +00:00
Ke Bao 16eb33ffe2 Update vocab embedding deps and add TP switch (#1856) 2024-10-31 20:13:07 -07:00