TensorRT-LLMs/tests/unittest/trt/model/redrafter
xiweny 6979afa6f2
test: reorganize tests folder hierarchy (#2996)
1. move TRT path tests to 'trt' folder
2. optimize some import usage
2025-03-27 12:07:53 +08:00
..
test_beams2tree.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_draft_token_indices.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_draft_token.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_gather_beams.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_mask.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_packed_position_ids.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_prefix_match_indices.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_prepare_input.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_process_logits.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_top1.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_unpack_gen_data.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_validate.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00