TensorRT-LLMs/examples/models/core
Ruoqian Guo db7446fda7
Feat: add deep_gemm swapab Kernel (#4430)
* feat: add deepgemm_swapab

feat: add fp8_gemm_kernel_swapab

Signed-off-by: Ruoqian Guo <ruoqiang@nvidia.com>

feat: set threshold for deepgemm and deepgemmswapab

Signed-off-by: Ruoqian Guo <ruoqiang@nvidia.com>

* docs: update README.md

Signed-off-by: Ruoqian Guo <ruoqiang@nvidia.com>

* fix: std::runtime_error needs #include <stdexcept>

Signed-off-by: Ruoqian Guo <ruoqiang@nvidia.com>

* chores: remove the redundant code

Signed-off-by: Ruoqian Guo <ruoqiang@nvidia.com>

* feat: support for dense deep_gemm swapab

Signed-off-by: Ruoqian Guo <ruoqiang@nvidia.com>

* chores: remove redundant code

Signed-off-by: Ruoqian Guo <ruoqiang@nvidia.com>

---------

Signed-off-by: Ruoqian Guo <ruoqiang@nvidia.com>
Co-authored-by: Tao Li @ NVIDIA <tali@nvidia.com>
2025-05-21 10:48:43 +08:00
..
bert Adding local paths to the datasets to make them loadable in offline mode (#3750) 2025-04-25 03:51:00 +08:00
commandr doc: fix path after examples migration (#3814) 2025-04-24 02:36:45 +08:00
deepseek_v3 Feat: add deep_gemm swapab Kernel (#4430) 2025-05-21 10:48:43 +08:00
enc_dec chore: Remove deprecated Python runtime benchmark (#4171) 2025-05-14 18:41:05 +08:00
exaone doc: fix path after examples migration (#3814) 2025-04-24 02:36:45 +08:00
gemma doc: fix path after examples migration (#3814) 2025-04-24 02:36:45 +08:00
glm-4-9b doc: fix path after examples migration (#3814) 2025-04-24 02:36:45 +08:00
gpt move the reset models into examples/models/core directory (#3555) 2025-04-19 20:48:59 -07:00
granite move the reset models into examples/models/core directory (#3555) 2025-04-19 20:48:59 -07:00
internlm2 doc: fix path after examples migration (#3814) 2025-04-24 02:36:45 +08:00
llama doc: fix path after examples migration (#3814) 2025-04-24 02:36:45 +08:00
mamba move the reset models into examples/models/core directory (#3555) 2025-04-19 20:48:59 -07:00
mixtral move the reset models into examples/models/core directory (#3555) 2025-04-19 20:48:59 -07:00
mllama doc: fix path after examples migration (#3814) 2025-04-24 02:36:45 +08:00
multimodal refactor: use x is None instead of x == None. (#4244) 2025-05-15 20:00:04 +08:00
nemotron move the reset models into examples/models/core directory (#3555) 2025-04-19 20:48:59 -07:00
nemotron_nas doc: fix path after examples migration (#3814) 2025-04-24 02:36:45 +08:00
phi move the reset models into examples/models/core directory (#3555) 2025-04-19 20:48:59 -07:00
qwen Breaking change: perf: Enable scheduling overlap by default (#4174) 2025-05-15 14:27:36 +08:00
qwen2audio move the reset models into examples/models/core directory (#3555) 2025-04-19 20:48:59 -07:00
qwenvl refactor: use x is None instead of x == None. (#4244) 2025-05-15 20:00:04 +08:00
recurrentgemma move the reset models into examples/models/core directory (#3555) 2025-04-19 20:48:59 -07:00
vit doc: fix path after examples migration (#3814) 2025-04-24 02:36:45 +08:00
whisper move the reset models into examples/models/core directory (#3555) 2025-04-19 20:48:59 -07:00