Baizhou Zhang fc91d08a8f [Revision] Add fast decode plan for flashinfer mla (#4012)

2025-03-05 11:20:41 -08:00

..

attention

[Revision] Add fast decode plan for flashinfer mla (#4012)

2025-03-05 11:20:41 -08:00

moe

Fix the moe padding conditional logic (#4081)

2025-03-05 10:56:51 -08:00

quantization

ROCM: AITER BLOCK GEMM (#4075)

2025-03-05 03:10:49 -08:00

activation.py

fix: support gelu_new activation function in gpt2 (#3712)

2025-03-04 04:09:52 -08:00

dp_attention.py

Support penalty in overlap mode; return logprob with chunked prefill; improve benchmark scripts (#3988)

2025-03-03 00:12:04 -08:00

layernorm.py

Support penalty in overlap mode; return logprob with chunked prefill; improve benchmark scripts (#3988)

2025-03-03 00:12:04 -08:00

linear.py

Support penalty in overlap mode; return logprob with chunked prefill; improve benchmark scripts (#3988)

2025-03-03 00:12:04 -08:00

logits_processor.py

[QUANT] Add GPTQModel Dynamic Quantization + lm_head Quantization (#3790)

2025-03-05 01:11:00 -08:00

parameter.py

Improve weight loading and code style (#3174)

2025-01-27 03:00:41 -08:00

pooler.py

Rename InputMetadata -> ForwardBatch (#1543)

2024-09-30 02:41:11 -07:00

radix_attention.py

[QUANT] Add GPTQModel Dynamic Quantization + lm_head Quantization (#3790)

2025-03-05 01:11:00 -08:00

rotary_embedding.py

Support penalty in overlap mode; return logprob with chunked prefill; improve benchmark scripts (#3988)

2025-03-03 00:12:04 -08:00

sampler.py

Support penalty in overlap mode; return logprob with chunked prefill; improve benchmark scripts (#3988)

2025-03-03 00:12:04 -08:00

torchao_utils.py

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

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

vocab_parallel_embedding.py

[QUANT] Add GPTQModel Dynamic Quantization + lm_head Quantization (#3790)

2025-03-05 01:11:00 -08:00