Yuan Luo 590f2da052 [Feat] Support Torch Symm Mem AllReduce (#10571) ...

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>

2025-10-05 13:55:19 -07:00

..

attention

[fix]enable flashmla when using draft model P/D attention select (#11012)

2025-10-04 20:59:34 +08:00

moe

Tiny fix ep_gather behavior different in CI (#11130)

2025-10-02 21:55:53 +08:00

quantization

Support single batch overlap (#10422)

2025-10-02 18:04:36 +08:00

activation.py

[Auto Sync] Update activation.py, chunk_cache.py, utils.py (20250917) (#10538)

2025-09-16 17:06:43 -07:00

amx_utils.py

[CPU] refine CPU integration code (#7647)

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

communicator.py

[Ascend]optimize Qwen3 on Ascend (#10574)

2025-09-22 17:18:36 -07:00

dp_attention.py

[Feat] Support Torch Symm Mem AllReduce (#10571)

2025-10-05 13:55:19 -07:00

elementwise.py

[Auto Sync] Update elementwise.py (20250923) (#10823)

2025-09-23 13:50:22 -07:00

flashinfer_comm_fusion.py

Support trtllm_allreduce_fusion in flashinfer for cuda<12.8 (#9339)

2025-08-20 16:54:30 -07:00

layernorm.py

Replace os.environ in layernorm.py (#10684)

2025-09-20 00:20:33 -07:00

linear.py

fix: fp8 quantization failure of qwen 2.5 VL 7B model (#10112)

2025-09-27 05:05:23 +00:00

logits_processor.py

Enable optional FP32 compute for LM Head (#10729)

2025-09-29 20:45:17 -07:00

model_parallel.py

Move parsers under a single folder (#9912)

2025-09-02 18:25:04 -07:00

multimodal.py

[VLM] Improving multimodal tensor hash kernel (#9008)

2025-08-14 13:45:55 -07:00

parameter.py

fix: fp8 quantization failure of qwen 2.5 VL 7B model (#10112)

2025-09-27 05:05:23 +00:00

pooler.py

Feat/support rerank (#6058)

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

radix_attention.py

Update grok.py and tiktoken tokenizer (#9532)

2025-08-23 05:40:18 -07:00

rocm_linear_utils.py

Optimized deepseek-v3/r1 model performance on mxfp4 run (#10008)

2025-09-04 15:11:22 -07:00

rotary_embedding.py

Quick Fix: fix Qwen3-VL launch failure caused by MRotaryEmbedding arg (#10985)

2025-09-30 22:17:05 -07:00

sampler.py

[2/2] Support deterministic inference for temperature > 0 (#10678)

2025-09-21 19:36:08 -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

fix: fp8 quantization failure of qwen 2.5 VL 7B model (#10112)

2025-09-27 05:05:23 +00:00

vocab_parallel_embedding.py

Revert "Revert the changes on NCCL symmetric memory" (#10238)

2025-09-09 12:11:49 -07:00