TensorRT-LLMs/tests/unittest/_torch
Michal Guzek 8e9409ce04
[https://nvbugs/5628204][fix] Stop token IDs - fast path optimization for single stop token IDs only (#9014)
Signed-off-by: Michal Guzek <mguzek@nvidia.com>
Signed-off-by: Michal Guzek <moraxu@users.noreply.github.com>
2025-11-13 14:17:20 +01:00
..
attention [https://nvbugs/5453806][unwaive] Unwaive fp8 kvcache attention test (#7243) 2025-09-05 12:13:57 -04:00
auto_deploy [https://nvbugs/5606166][fix] AutoDeploy: use tuples for cudagraph shape lookup (#8772) 2025-10-31 13:59:48 +01:00
compilation [TRTLLM-3105][feat] Add Piecewise CUDA Graph Support (#3804) 2025-05-09 11:04:01 +08:00
debugger Fix: fix nvbug 5356427 (#5464) 2025-06-25 22:24:26 +08:00
executor [TRTLLM-8650][fix] beam search request validation (#8433) 2025-10-21 10:50:27 +02:00
misc [TRTLLM-4629] [feat] Add support of CUDA13 and sm103 devices (#7568) 2025-09-16 09:56:18 +08:00
modeling [TRTLLM-8658][infra] upgrade to DLFW 25.10 and pytorch 2.9.0 / triton 3.5.0 (#8621) 2025-11-03 09:24:58 +08:00
models/checkpoints/hf [None][feat] Skip prefetching consolidated safetensors when appropriate (#7013) 2025-08-25 23:56:21 -04:00
modules [https://nvbugs/5467531][fix] Fix moe test and wide ep fake impl (#8883) 2025-11-06 11:40:50 +08:00
multi_gpu [https://nvbugs/5597647][fix] Fix MNNVL unit test failed due to accuracy issue on Hopper (#8891) 2025-11-06 18:28:06 +01:00
multi_gpu_modeling [https://nvbugs/5515753][ci] Add NCCL_DEBUG=INFO flag to collect more info with CI failure. (#8440) 2025-10-21 18:12:05 -07:00
multimodal [None][ci] Waive test_mm_encoder_standalone.py::test_multi_request_batch_chat[llava-v1.6-mistral-7b-hf] (#8010) 2025-09-26 11:07:54 +08:00
sampler [https://nvbugs/5628204][fix] Stop token IDs - fast path optimization for single stop token IDs only (#9014) 2025-11-13 14:17:20 +01:00
speculative [https://nvbugs/5537878][fix] Reserve an extra slot for padded batch … (#8231) 2025-10-13 23:34:22 -07:00
thop [None][infra] Waive failed tests for release branch on 11/03 (#8879) 2025-11-03 02:59:33 -08:00
helpers.py [TRTLLM-7330][feat] Eagle3 cuda graph support for the first draft model inference (#7363) 2025-09-26 11:28:05 +08:00
pattern_watcher.py [TRTLLM-3105][feat] Add Piecewise CUDA Graph Support (#3804) 2025-05-09 11:04:01 +08:00
test_connector.py [None][feat] KV Cache Connector API (#7228) 2025-08-28 23:09:27 -04:00