TensorRT-LLMs/tensorrt_llm/_torch/auto_deploy
Stefan Niebler d1112aac37
[TRTLLM-3442] feat: added beam search support to the PyTorch Workflow (#5333)
Signed-off-by: Stefan Niebler <82932102+stnie@users.noreply.github.com>
2025-07-05 01:35:13 +09:00
..
compile [AutoDeploy]feat: Add an AutoDeploy compile backend that only calls torch.compile (#4240) 2025-05-16 08:38:15 +08:00
custom_ops feat: AutoDeploy fp8 quantization support for bmm (#3849) 2025-06-30 12:36:34 -04:00
distributed Use backend to replace macro to control enablement of MNNVL all reduce (#4635) 2025-06-12 11:22:49 +08:00
models [AutoDeploy] merge feat/ad-2025-06-29 (#5737) 2025-07-04 10:21:18 +09:00
shim [TRTLLM-3442] feat: added beam search support to the PyTorch Workflow (#5333) 2025-07-05 01:35:13 +09:00
transformations [AutoDeploy] merge feat/ad-2025-06-29 (#5737) 2025-07-04 10:21:18 +09:00
utils [feat] Support torch compile for attention dp (#5086) 2025-07-01 13:48:52 -04:00
__init__.py [AutoDeploy] merge feat/ad-2025-06-13 (#5556) 2025-06-29 03:52:14 +08:00
llm_args.py [AutoDeploy] merge feat/ad-2025-06-29 (#5737) 2025-07-04 10:21:18 +09:00
llm.py [AutoDeploy] merge feat/ad-2025-06-13 (#5556) 2025-06-29 03:52:14 +08:00