Cheng Wan 9effeb5bdd Support EPLB in FusedMoE (#8448)

2025-07-29 16:02:41 -07:00

..

attention

[feat] Support different attention backends for prefill and decode (#6338)

2025-07-28 11:42:29 +08:00

moe

Support EPLB in FusedMoE (#8448)

2025-07-29 16:02:41 -07:00

quantization

Reduce memory usage for fp4 moe (#8413)

2025-07-28 22:51:23 -07:00

activation.py

[AMD] Add silu_and_mul, gelu_and_mul, gelu_tanh_and_mul, and gelu_quick kernels for AMD GPUs (#7135)

2025-07-24 23:44:28 -07:00

amx_utils.py

[CPU] refine CPU integration code (#7647)

2025-07-03 09:51:09 -07:00

communicator.py

DP Enhancement (#8280)

2025-07-24 21:36:21 -07:00

dp_attention.py

DP Enhancement (#8280)

2025-07-24 21:36:21 -07:00

elementwise.py

Improve streaming, log_level, memory report, weight loading, and benchmark script (#7632)

2025-06-29 23:16:19 -07:00

flashinfer_comm_fusion.py

Fix illegal memory in trtllm allreduce fusion (#7864)

2025-07-08 11:47:17 -07:00

layernorm.py

model: support intern-s1 (#8350)

2025-07-26 13:48:51 -07:00

linear.py

[1/N] MoE Refactor: refactor select_experts (#7966)

2025-07-19 00:51:15 -07:00

logits_processor.py

Save peak memory in logits processor (#8343)

2025-07-26 01:46:42 -07:00

multimodal.py

Support precomputed_embeddings for Llama 4 (#8156)

2025-07-27 01:14:49 -07:00

parameter.py

[CPU][Qwen3 MoE] Enable fused_topk CPU fusion and enhance FP8 TP padding (#7838)

2025-07-09 02:04:21 -07:00

pooler.py

Feat/support rerank (#6058)

2025-06-16 10:50:01 -07:00

radix_attention.py

DP Enhancement (#8280)

2025-07-24 21:36:21 -07:00

rotary_embedding.py

Ascend attention backend(PA&MLA) (#7722)

2025-07-03 09:23:19 -07:00

sampler.py

Reintroduce tiny fix sampler error when prob is not contiguous (#7354)

2025-06-19 09:39:34 -07:00

torchao_utils.py

Support loading of larger models with on-the-fly quantization (#3061)

2025-01-22 21:33:17 -08:00

utils.py

[Feature] Support Flashinfer fmha on Blackwell (#6930)

2025-06-06 12:57:50 -07:00

vocab_parallel_embedding.py

Refactor: move all quantization-related code to srt/layer/quantization (#7989)

2025-07-17 00:47:07 -07:00