Yukun He ff82aef99b Fix the issues related to fused moe path. (#3435) ... * One of the tactic is not supported during dispatch. * final_hidden_states should be unpacked if it is not min_latency_mode. Signed-off-by: Yukun He <23156053+hyukn@users.noreply.github.com>		2025-04-11 21:41:15 +08:00
..
__init__.py	Update TensorRT-LLM (#2755)	2025-02-11 03:01:00 +00:00
attention.py	[MLA] Deallocate tensors after use (#3286)	2025-04-09 21:36:07 -07: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	Fix the issues related to fused moe path. (#3435)	2025-04-11 21:41:15 +08:00
gated_mlp.py	lora_tests (#3201)	2025-04-09 18:06:52 +03:00
linear.py	lora_tests (#3201)	2025-04-09 18:06:52 +03: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	lora_tests (#3201)	2025-04-09 18:06:52 +03:00
rms_norm.py	Add Llama 4 (#3302)	2025-04-09 03:35:21 +08:00
rotary_embedding.py	feat: support llama4 nope layers; support FP8 checkpoint loading; (#3382)	2025-04-10 10:16:42 -07:00