TensorRT-LLMs/tests/unittest/_torch/speculative
2025-07-02 04:54:43 -04:00
..
test_draft_target.py [TRTLLM-5530][BREAKING CHANGE]: enhance the llm args pytorch config part 1(cuda_graph_config) (#5014) 2025-06-30 11:05:40 +08:00
test_eagle3.py [https://nvbugspro.nvidia.com/bug/5329655] [feat] Pytorch path add spec dec param to attention op (#5146) 2025-07-02 04:54:43 -04:00
test_mtp.py fix: refactor and fix mtp vanilla (#4762) 2025-06-20 05:23:39 +08:00
test_ngram.py [TRTLLM-5530][BREAKING CHANGE]: enhance the llm args pytorch config part 1(cuda_graph_config) (#5014) 2025-06-30 11:05:40 +08:00