TensorRT-LLMs/tests/unittest/_torch/multi_gpu
liji-nv e0d0dde058
None - Add one-shot version for UB AR NORM FP16/BF16 (#2995)
Signed-off-by: Jin Li <59594262+liji-nv@users.noreply.github.com>
2025-03-31 11:16:03 +08:00
..
test_ar_residual_norm.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_deepseek_allreduce.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_embedding.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_linear.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_star_attention_input.jsonl Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_star_attention.py move BuildConfig functional args to llmargs (#3036) 2025-03-29 02:20:18 +08:00
test_user_buffers.py None - Add one-shot version for UB AR NORM FP16/BF16 (#2995) 2025-03-31 11:16:03 +08:00