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
17e0d0fb1a
TensorRT-LLMs/tensorrt_llm/_torch/modules/mamba
History
tomeras91 7dbecf7272
[TRTLLM-4923][feat] Enable CUDA graphs for Nemotron-H (#5646) 
Signed-off-by: Tomer Asida <57313761+tomeras91@users.noreply.github.com>
2025-07-03 11:07:51 +03:00
..
__init__.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
causal_conv1d.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
layernorm_gated.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
mamba2_metadata.py [TRTLLM-5835][feat] Optimized Mamba2Mixer prefill (#5128) 2025-06-16 16:29:17 +03:00
mamba2_mixer.py [TRTLLM-4923][feat] Enable CUDA graphs for Nemotron-H (#5646) 2025-07-03 11:07:51 +03:00
selective_state_update.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
softplus.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
ssd_bmm.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
ssd_chunk_scan.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
ssd_chunk_state.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
ssd_combined.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00
ssd_state_passing.py [TRTLLM-4783][feat] Mamba2 kernel updates for Nemotron-H (#4494) 2025-06-01 13:56:44 +03:00