Liangsheng Yin 73d4a5f879 Organize spec-related data structures (#10735)

2025-10-01 09:45:30 +08:00

..

attention

Organize spec-related data structures (#10735)

2025-10-01 09:45:30 +08:00

moe

Organize spec-related data structures (#10735)

2025-10-01 09:45:30 +08:00

quantization

[model] added support for w8a8int8 used by neuralmagic/Qwen2-0.5B-Ins… (#9642)

2025-09-30 11:26:17 +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

Temporay work-around for rocm 7.0.0 alpha with enabling data-parallel issue (#10434)

2025-09-15 19:08:04 -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

Fix FusedSetKVBufferArg in RotaryEmbedding (#11003)

2025-09-28 11:17:27 -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