Jinyang Yuan 992d513bc6 feat: Optionally split MoE inputs into chunks to reduce GPU memory usage (#3104) ... Signed-off-by: Jinyang Yuan <154768711+jinyangyuan-nvidia@users.noreply.github.com> Co-authored-by: raccoonliukai <raccoonliu@tencent.com>		2025-04-01 16:07:02 +08:00
..
attention_backend	fix：fix illeagel memory access when mtp >= 2 (#3006)	2025-04-01 13:36:45 +08:00
auto_deploy	perf: [AutoDeploy] Enable AutoDeploy as a backend in trtllm-bench (#3041)	2025-03-26 14:33:14 -07:00
compilation	None - Add one-shot version for UB AR NORM FP16/BF16 (#2995)	2025-03-31 11:16:03 +08:00
custom_ops	Update (#2978)	2025-03-23 16:39:35 +08:00
models	feat: Optionally split MoE inputs into chunks to reduce GPU memory usage (#3104)	2025-04-01 16:07:02 +08:00
modules	feat: Optionally split MoE inputs into chunks to reduce GPU memory usage (#3104)	2025-04-01 16:07:02 +08:00
pyexecutor	feat: Optionally split MoE inputs into chunks to reduce GPU memory usage (#3104)	2025-04-01 16:07:02 +08:00
speculative	fix：fix illeagel memory access when mtp >= 2 (#3006)	2025-04-01 13:36:45 +08:00
__init__.py	Update TensorRT-LLM (#2755)	2025-02-11 03:01:00 +00:00
autotuner.py	Update (#2978)	2025-03-23 16:39:35 +08:00
distributed.py	Update (#2978)	2025-03-23 16:39:35 +08:00
llm.py	test: [TRTLLM-4334] Create 1.0 criteria scope from API stability references (#3069)	2025-03-26 18:14:35 +08:00
metadata.py	Update TensorRT-LLM (#2936)	2025-03-18 21:25:19 +08:00
model_config.py	feat: Optionally split MoE inputs into chunks to reduce GPU memory usage (#3104)	2025-04-01 16:07:02 +08:00
pipeline_interface.py	Update (#2978)	2025-03-23 16:39:35 +08:00
utils.py	feat: Optionally split MoE inputs into chunks to reduce GPU memory usage (#3104)	2025-04-01 16:07:02 +08:00