HandH1998 2ac189edc8 Amd test fp8 (#4261)

2025-03-10 10:12:09 -07:00

..

attention

Simplify tests & Fix trtllm custom allreduce registration (#4252)

2025-03-10 01:24:22 -07:00

moe

Fix quantization and nightly tests (#4258)

2025-03-10 03:06:21 -07:00

quantization

Amd test fp8 (#4261)

2025-03-10 10:12:09 -07: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

Apply sgl w8a8 fp8 kernel (#3148)

2025-03-09 00:03:32 -08:00

logits_processor.py

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

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

parameter.py

Apply sgl w8a8 fp8 kernel (#3148)

2025-03-09 00:03:32 -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

[EAGLE] many fixes for eagle (#4195)

2025-03-07 22:12:13 -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

Clean up fp8 support (#4230)

2025-03-09 21:46:35 -07:00