kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
2a2b7bfc66
TensorRT-LLMs/tensorrt_llm/_torch/modules
History
Tracin 2a2b7bfc66
Fix miss bias add for FP4Linear. (#3361) 
Signed-off-by: Tracin <10434017+Tracin@users.noreply.github.com>
Co-authored-by: QI JUN <22017000+QiJune@users.noreply.github.com>
2025-04-09 09:17:54 +08:00
..
__init__.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
attention.py Add Llama 4 (#3302) 2025-04-09 03:35:21 +08:00
decoder_layer.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
embedding.py Update (#2978) 2025-03-23 16:39:35 +08:00
fused_moe.py Add Llama 4 (#3302) 2025-04-09 03:35:21 +08:00
gated_mlp.py perf: Add optimizations for deepseek in min latency mode (#3093) 2025-04-02 09:05:24 +08:00
linear.py Fix miss bias add for FP4Linear. (#3361) 2025-04-09 09:17:54 +08:00
logits_procesor.py Update (#2978) 2025-03-23 16:39:35 +08:00
mamba.py Update TensorRT-LLM (#2820) 2025-02-25 21:21:49 +08:00
mlp.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
rms_norm.py Add Llama 4 (#3302) 2025-04-09 03:35:21 +08:00
rotary_embedding.py Add Llama 4 (#3302) 2025-04-09 03:35:21 +08:00